mindspore.ops.size

mindspore.ops.size(input_x)

计算输入tensor的元素总数。

参数：

• input_x (Tensor) - 输入tensor。

返回：

int

支持平台：

Ascend GPU CPU

样例：

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