torch.sqrt()的使用

参考链接: torch.sqrt(input, out=None)

功能: 
	逐元素计算张量的平方根

代码实验:

(base) PS C:\Users\chenxuqi> python
Python 3.7.4 (default, Aug  9 2019, 18:34:13) [MSC v.1915 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>>
>>> torch.manual_seed(seed=20200910)
<torch._C.Generator object at 0x0000019723A22FD0>
>>> a = torch.randn(4)
>>> a
tensor([ 0.2824, -0.3715,  0.9088, -1.7601])
>>> torch.sqrt(a)
tensor([0.5314,    nan, 0.9533,    nan])
>>> a.sqrt()
tensor([0.5314,    nan, 0.9533,    nan])
>>>
>>>