mindspore.ops.Sqrt
- class mindspore.ops.Sqrt[源代码]
逐元素计算输入Tensor的平方根。
Note
当输入数据存在一些负数,则负数对应位置上的返回结果为NaN。
\[out_{i} = \sqrt{x_{i}}\]- 输入:
x (Tensor) - Sqrt的输入,数据类型为数值型,shape: \((N, *)\) ,其中 \(*\) 表示任意数量的附加维度。其秩应小于8。
- 输出:
Tensor,shape和数据类型与输入 x 相同。
- 异常:
TypeError - x 不是Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> x = Tensor(np.array([1.0, 4.0, 9.0]), mindspore.float32) >>> sqrt = ops.Sqrt() >>> output = sqrt(x) >>> print(output) [1. 2. 3.]