Количество элементов в многомерном массиве Python

import numpy as np
size = 1
for dim in np.shape(a): size *= dim
Xerothermic Xenomorph