Pytorch torch.rsqrt()的简单用法

官方文档：https://pytorch.org/docs/master/generated/torch.rsqrt.html

简单来说就是对每个元素取平方根后再取倒数，即：
$${\text{ out }}_i=\frac{1}{\sqrt{{\text{ input }}_i}}$$
是元素级的操作，并不会影响如维度等因素。

例如：

import torch

x = torch.Tensor([[1, 4],[16, 25]])
y = torch.rsqrt(x)
print(y)

输出：

tensor([[1.0000, 0.5000],
        [0.2500, 0.2000]])