jax.numpy.size

jax.numpy.size(a, axis=None)

傳回沿著指定軸的元素數量。

參數:

• a (array_like) – 輸入資料。

• axis (int, optional) – 沿著要計算元素的軸。預設為元素總數。

傳回值:

element_count – 沿著指定軸的元素數量。

傳回類型:

int

另請參閱

shape

陣列的維度

ndarray.shape

陣列的維度

ndarray.size

陣列中的元素數量

範例

>>> import numpy as np
>>> a = np.array([[1,2,3],[4,5,6]])
>>> np.size(a)
6
>>> np.size(a,1)
3
>>> np.size(a,0)
2