LayerNorm — PyTorch 1.10.1 documentation
pytorch.org › docs › stableThe mean and standard-deviation are calculated over the last D dimensions, where D is the dimension of normalized_shape.For example, if normalized_shape is (3, 5) (a 2-dimensional shape), the mean and standard-deviation are computed over the last 2 dimensions of the input (i.e. input.mean((-2,-1))).