10.TensorRT中文版开发教程-----TensorRT中的循环
TensorRT中的循环点击此处加入NVIDIA开发者计划NVIDIA TensorRT 支持循环结构,这对于循环网络很有用。 TensorRT 循环支持扫描输入张量、张量的循环定义以及“扫描输出”和“最后一个值”输出。10.1. Defining A Loop循环由循环边界层(loop boundary layers)定义。ITripLimitLayer指定循环迭代的次数。IIteratorLayer使循环能够迭代张量。IRecurrenceLayer指定一个循环定义。ILoopOut