torch.rsqrt()函数详解

torch.rsqrt(input, *, out=None) 函数

• torch.rsqrt(input, *, out=None)

• 数学公式：

  针对输入input的每个元素的平方根的倒数来返回一个新的Tensor。
  $${\text{out}}_i=\frac{1}{\sqrt{{\text{input}}_i}}$$

• # example
  >>> a = torch.randn(4)
  >>> a
  tensor([-0.0370,  0.2970,  1.5420, -0.9105])
  >>> torch.rsqrt(a)
  tensor([    nan,  1.8351,  0.8053,     nan])