mindspore.ops.Square

class mindspore.ops.Square[源代码]

计算输入Tensor的平方。

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

输入:

  • x (Tensor) - Square的输入,其数据类型为数值型,shape: \((N, *)\) ,其中 \(*\) 表示任意数量的附加维度。其秩应小于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.]