mindspore.mint.rsqrt

mindspore.mint.rsqrt(input)[源代码]

逐元素计算输入Tensor元素的平方根倒数。

\[out_{i} = \frac{1}{\sqrt{input_{i}}}\]
参数:
  • input (Tensor) - rsqrt的输入Tensor,其每个元素都为非负,若某个元素为负,计算结果为nan。

返回:

Tensor,具有与 input 相同的shape。

异常:
  • TypeError - 如果 input 不是Tensor。

支持平台:

Ascend

样例:

>>> import mindspore as ms
>>> from mindspore import mint
>>> input = ms.Tensor([-0.0370,  0.2970,  1.5420, -0.9105])
>>> output = mint.rsqrt(input)
>>> print(output)
[       nan 1.8349396  0.8053002        nan]