torch.sqrt¶ torch.sqrt(input, *, out=None) → Tensor¶ 返回一个新的张量,其中包含 input 元素的平方根。 outi=inputi\text{out}_{i} = \sqrt{\text{input}_{i}} outi=inputi 参数 input (Tensor) – 输入张量。 关键字参数 out (Tensor, 可选) – 输出张量。 示例 >>> a = torch.randn(4) >>> a tensor([-2.0755, 1.0226, 0.0831, 0.4806]) >>> torch.sqrt(a) tensor([ nan, 1.0112, 0.2883, 0.6933])