【TensorRT】TensorRT学习笔记之网络定义与构建

一、网络定义的核心接口

1. ​创建Logger

   • ​功能：日志记录器，用于捕获构建过程中的错误、警告和调试信息。
   • ​实现示例：

     class Logger : public nvinfer1::ILogger {
         void log(Severity severity, const char* msg) noexcept override {
             if (severity <= Severity::kWARNING)
                 std::cout << "[TensorRT] " << msg << std::endl;
         }
     } gLogger;  // 单例模式

   • ​注意：必须传递给所有TensorRT对象（如Builder、Parser）。

2. ​构建器（IBuilder）与网络定义（INetworkDefinition）​

   • ​初始化：

     nvinfer1::IBuilder* builder = createInferBuilder(gLogger);
     uint32_t flag = 1U <&l