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.]