Solved by vainaijr in post #8 
              
              
                I think doing 
x = torch.randn(1, 3, 6) # batch size 1, 3 channels, 6 length of sequence
a = nn.Conv1d(3, 6, 3) # in channels 3, out channels 6, kernel size 3
gn = nn.GroupNorm(1, 6)
gn(a(x))

tensor([[[-0.1459,  0.5860,  0.1771,  1.1413], 
[-0.8613,  2.7552, -1.0135,  0.8898], 
[-0.1119, -0.1656, -…