mindspore.Tensor.square

mindspore.Tensor.square()

逐元素计算 self 的平方。

\[out_{i} = self_{i} ^ 2\]

返回：

Tensor，数据类型和shape与 self 相同。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor
>>> input = Tensor(np.array([1.0, 2.0, 3.0]), mindspore.float32)
>>> output = Tensor.square(input)  # input.square()
>>> print(output)
[1. 4. 9.]