jax.numpy.size

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

返回给定轴上的元素数量。

参数:

• a (类数组) – 输入数据。

• axis (int, 可选) – 计算元素的轴。 默认情况下，给出元素总数。

返回:

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