title: PyTorch BatchNorm1D, 2D, 3D and TensorFlow/Keras BatchNormalization
body: <p>I want to use <code>BatchNorm1D</code> like in PyTorch in TensorFlow. I notice that <code>BatchNormalization()</code> in TF has <code>axis=-1</code> as default. Which axis is the correct one for <code>BatchNorm1D, BatchNorm2D, BatchNorm3D</code> as in PyTorch?</p>