Yes, an executable code snippet would still be needed, as I cannot reproduce the issue via:
# CPU
input = torch.randn(20, 5, 10, 10)
m = nn.LayerNorm(input.size()[1:])
output = m(input)

# GPU
input = input.cuda()
m.cuda()
output = m(input)