通过在模型的前向传播方法中使用nn.BatchNorm2d层,我们可以将输入数据进行归一化处理。 数据归一化的注意事项 在使用PyTorch的数据归一化层时,有一些注意事项需要考虑: 数据归一化层一般放置在卷积层或全连接层之后,激活函数之前。 归一化层的输入通常是一个四维张量,表示为(batch_size, channels, height, width)。