I think there’s an error in the documentation for BatchNorm1d 2. It currently says that num_features is “L from input of size (N,L)”. Shouldn’t it say “C from input of size (N,C)”, as the input shape specification correctly states?
Also, should BatchNorm0d not have its own function? The current API seems confusing since it doesn’t fit the expected pattern of

BatchNorm3d for inputs of shape of shape [C,D,H,W]
BatchNorm2d for inputs of shape [C,H,W]
BatchNorm1d for inputs of shape [C,L]
BatchNorm0d for inputs of shape [C]