mindspore.ops.Rsqrt
- class mindspore.ops.Rsqrt[源代码]
逐元素计算输入Tensor平方根的倒数。
\[out_{i} = \frac{1}{\sqrt{x_{i}}}\]- 输入:
x (Tensor) - Rsqrt的输入,其rank必须在[0, 7]内,并且每个元素必须是非负数。
- 输出:
Tensor,数据类型和shape与 x 相同。
- 异常:
TypeError - 如果 x 不是Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> input_tensor = Tensor([[4, 4], [9, 9]], mindspore.float32) >>> rsqrt = ops.Rsqrt() >>> output = rsqrt(input_tensor) >>> print(output) [[0.5 0.5 ] [0.33333334 0.33333334]]