I have a 3-dimensional array:
A = np.random.randint(0, 1000, (5, 4, 3))
Output:
array([[[101, 131, 271], [565, 497, 30],