slavavs:

out = out.contiguous (). View (1,1, batch, wn1) -> self.bn1 = nn.BatchNorm2d (batch, affine = True)


sorry
2. out = out.contiguous (). View (1,1, batch, wn1) -> self.bn1 = nn.BatchNorm2d (1, affine = True)