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