Return axis number(s) corresponding to dimension(s) in this array.
dim : str or iterable of str
Dimension name(s) for which to lookup axes.
int or tuple of int
Axis number or numbers corresponding to the given dimensions.