mindspore.ops.numel

mindspore.ops.numel(input)

返回tensor的元素的总数量。

参数：

• input (Tensor) - 输入tensor。

返回：

Tensor元素总数量。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> input_x = mindspore.tensor([[2, 2], [2, 2]], mindspore.float32)
>>> print(mindspore.ops.numel(input_x))
4