Correct Usage of LayerNorm

1

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 ?