全连接层:nn.Linear
卷积层:nn.Conv2d:卷积中的bias
nn.Conv3d
池化层:nn.MaxPool2d
批量归一化:nn.BatchNorm2d
BN层理解
Pytorch详解nn模块
于 2022-01-18 21:06:16 首次发布
全连接层:nn.Linear
卷积层:nn.Conv2d:卷积中的bias
nn.Conv3d
池化层:nn.MaxPool2d
批量归一化:nn.BatchNorm2d
BN层理解