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