Assuming it is 2 dimensional:

np.random.default_rng().permuted(Xtrain, axis=1)

np.random.default_rng().permuted(np.arange(9).reshape((3,3)), axis=1)
Out[6]: 
array([[0, 2, 1],
       [3, 4, 5],
       [8, 6, 7]])