mindspore.ops.Size

class mindspore.ops.Size[source]

Returns the size of a tensor.

Returns an int scalar representing the elements size of input, the total number of elements in the tensor.

Inputs:
  • input_x (Tensor) - The shape of tensor is (x1,x2,...,xR). The data type is Number.

Outputs:

int. A scalar representing the elements size of input_x, tensor is the number of elements in a tensor, size=x1x2...xR. The data type is an int.

Raises

TypeError – If input_x is not a Tensor.

Supported Platforms:

Ascend GPU CPU

Examples

>>> input_x = Tensor(np.array([[2, 2], [2, 2]]), mindspore.float32)
>>> size = ops.Size()
>>> output = size(input_x)
>>> print(output)
4