mindspore.mint.square

mindspore.mint.square(input)

Return square of a tensor element-wise.

$$y_i=inpu{t}_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.]