官方文档地址 https://docs.nvidia.com/deeplearning/tensorrt/developer-guide/index.html#define-loops
非顺序结构,其内容确实有点乱,而且没有完整可运行的样例。
可以有多个IIteratorLayer, IRecurrenceLayer, and ILoopOutputLayer 层,最多有2个ITripLimitLayers层。
这里贴2个可运行的代码样例,分别是2种ITripLimitLayers层(TripLimit::kCOUNT 、 TripLimit::kWHILE),简单可运行的代码以帮助入门:
trt.TripLimit.COUNT
loop.add_trip_limit(trip_limit.get_output(0), trt.TripLimit.COUNT)
trt.TripLimit.WHILE
实现类似于for(i = 0; i<3;i++)
i_init = network.add_constant(shape=(), weights=trt.Weights(np.array([0], dtype=np.dtype("i"))))
i_one = network.add_constant(shape=(), weights=trt.Weights(np.array([1], dtype=np.d