torch.rsqrt()

官方解释: Link

函数的作用：获取 input 张量中每一个元素的 平方根的倒数。

o u t p u t i = 1 i n p u t i output_i = \frac{1}{\sqrt{input_i}} outputi​=inputi​ ​1​

import torch
a = torch.tensor([-0.0370,  0.2970,  1.5420, -0.9105, 0])
res = a.rsqrt()         或者写成 res = torch.rsqrt(a) 
res

#输出 
tensor([   nan, 1.8349, 0.8053,    nan,    inf])