Pytorch转TensorRT中的坑
Pytorch转ONNX再转TensorRT，其中遇到一些转的时候会出现的层需要修改的问题，这里对修改的层做一些总结。reshape Pytorch中会有很多需要reshape的操作，可用的算子有：reshapeviewflatten 前两个都是需要指定reshape后完整的尺寸大小，因此使用中需要先获取输入数据的维度，这个在Pytorch框架下使用没有问题，但用到Tens...
