It's simple. It's the axis that will disappear (or become length 1 with keepdims) after performing the operation.
Both of them are of interest and after you were confused once about which one to supply to axis=..., there is no way back to clear the confusion. With einsum there is no confusion.