Python PyTorch sqrt用法及代码示例

本文简要介绍python语言中 torch.sqrt 的用法。

用法:

torch.sqrt(input, *, out=None) → Tensor

参数：

input(Tensor) -输入张量。

关键字参数：

out(Tensor,可选的) -输出张量。

返回具有 input 元素平方根的新张量。

\text{out}_{i} = \sqrt{\text{input}_{i}}

例子：

>>> 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])