mindspore.ops.numel
- mindspore.ops.numel(input)[源代码]
返回Tensor的元素的总数量。
- 参数:
input (Tensor) - 输入Tensor。
- 返回:
int。Tensor的元素的总数量。
- 异常:
TypeError - input 不是Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> input_x = Tensor(np.array([[2, 2], [2, 2]]), mindspore.float32) >>> print(ops.numel(input_x)) 4