一、基础算子类
1. 卷积层(IConvolutionLayer
)
- 功能:执行2D/3D卷积操作。
- 关键参数:
IConvolutionLayer* conv = network->addConvolution( *input, // 输入张量(ITensor*) num_output_maps, // 输出通道数(int) DimsHW{kH, kW}, // 卷积核尺寸(nvinfer1::DimsHW) weights, // 卷积核权重(Weights类型) biases // 偏置(Weights类型,可选) );
- 成员函数:
setStride(DimsHW stride)
:设置步长(默认(1,1)
)。setPadding(DimsHW padding)
:设置对称填充(默认(0,0)
)。setDilation(DimsHW dilation)
:设置空洞率(默