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