Hi I want to use LayerNorm to apply per channel normalization for an image.
Assume a minibatch is of shape [B, C, H, W], I want to normalize across the C dimension. How to achive this use nn.LayerNorm class ?