Custom_TensorRT 这是一个代码制作自定义卷积层。（使用CUDA内核，TensorRT IPlugin） 您可以从mnist_torch_tensorrt.py获取重量文件。 MNIST Conv2D（5X5内核，零填充，OutChannel = 5）：（1、28、28）->（5、24、24） Maxpooling（2X2内核，零填充，2X2步幅）：（5，24，24）->（5，12，12） FClayer1：（5，12，12）->（120，） Relu