【TensorRT】TensorRT学习笔记之INetworkDefinition类

INetworkDefinition类是TensorRT中用于定义神经网络结构的核心类。以下是对INetworkDefinition类的详细说明：

一、概述

INetworkDefinition类在TensorRT中扮演着至关重要的角色，它负责创建和定义神经网络的结构。通过INetworkDefinition类，用户可以添加输入层、输出层以及中间的隐藏层，定义各层之间的连接关系，并设置网络的输入和输出张量。

二、成员函数详细说明

2.1. 输入/输出管理

• ​ITensor* addInput(const char* name, DataType type, Dims dimensions)
  添加网络输入张量，支持动态维度（如 Dims4{-1, 3, 224, 224}）。

• ​void markOutput(ITensor& tensor)
  将指定张量标记为网络输出。

2.​2. 层操作

• ​IConvolutionLayer* addConvolution(...)
  添加卷积层。
• ​IPoolingLayer* addPooling(...)
  添加池化层。