以上代码，哪个是批归一化层

在以上的代码中，`self.bn1` 和 `self.bn2` 是批归一化层。它们分别在第一个卷积层和第二个卷积层后面被调用。

具体来说，`self.bn1` 在第一个卷积层的输出经过激活函数后被调用，用于对输出进行批归一化。同样地，`self.bn2` 在第二个卷积层的输出经过激活函数后被调用，也用于对输出进行批归一化。

以下是代码片段中与批归一化层相关的部分：

self.bn1 = nn.BatchNorm2d(out_channels)
...
self.bn2 = nn.BatchNorm2d(out_channels)

这里使用的是 `nn.BatchNorm2d`，它是一个针对二维输入进行批归一化的函数。`out_channels` 是指定输出通道数的参数，它决定了每个通道上进行归一化的样本数。