Is this the same as np.rollaxis ?