9.TensorRT中文版开发教程-----TensorRT中的自定义层(Extending TensorRT With Custom Layers)

TensorRT中的自定义层点击此处加入NVIDIA开发者计划NVIDIA TensorRT 支持多种类型的层，其功能不断扩展；但是，在某些情况下，支持的层不能满足模型的特定需求。您可以通过实现自定义层（通常称为插件）来扩展 TensorRT。9.1. Adding Custom Layers Using The C++ API您可以通过从 TensorRT 的插件基类之一派生来实现自定义层。