mindspore.ops.Square

class mindspore.ops.Square[源代码]

计算输入Tensor的平方。

\[out_{i} = (x_{i})^2\]
输入：

  • x (Tensor) - Square的输入，其数据类型为数值型，其秩应小于8。

输出：

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

异常：

  • TypeError - 如果 x 不是Tensor。

支持平台：

Ascend GPU CPU

样例：

>>> x = Tensor(np.array([1.0, 2.0, 3.0]), mindspore.float32)
>>> square = ops.Square()
>>> output = square(x)
>>> print(output)
[1. 4. 9.]