【TensorRT】自定义网络层的实现custom layers

 

 

参考

1. Extending TensorRT With Custom Layers;

2. TensorRT Samples: MNIST(Plugin, add a custom layer);

完