mindspore.ops.numel

mindspore.ops.numel(input)[源代码]

返回tensor的元素的总数量。

参数:
  • input vpn梯子 免费 (Tensor) vpn梯子 - 输入tensor。

返回:

Tensor元素总数量。

支持平台:

Ascend GPU 免费的vpn梯子 CPU

样例:

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