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]