Solved by ptrblck in post #2 
              
              
                The last reshape operation will interleave the output values and you would need to permute y before calling reshape: 

hidden_channels = 4
bn1 = torch.nn.BatchNorm1d(hidden_channels)

x = torch.randn(2, 3, 4)
batch_size, num_nodes, num_channels = x.size()
y = x.clone()
x = x.view(-1, num_channels)
x…