I expected the following code to have True for all entries of a but it doesn’t. Why is that? (I’m trying to understand the difference between an input shape N,C,L and N,L for batch norm 1D)
  self.bn1 = torch.nn.BatchNorm1d(hidden_channels)
  def bn(self, x):
        batch_size, num_nodes, num_channels = x.size()
        y = x.clone()
        x = x.view(-1, num_channels)
        x = getattr(self, "bn1")(x)
        y = y.transpose(2,1)
        y = getattr(self,"bn1")(y)
        y = y.reshape(-1, num_channels)
        a = torch.eq(x,y)