mindspore.mint.square

mindspore.mint.square(input)[source]

Return square of a tensor element-wise.

\[y_i = input_i ^ 2\]

Parameters: input (Tensor) – The input tensor.
Returns: Tensor

Supported Platforms:: Ascend

Examples

>>> import mindspore
>>> input = mindspore.tensor([1.0, 2.0, 3.0], mindspore.float32)
>>> output = mindspore.mint.square(input)
>>> print(output)
[1. 4. 9.]