Python PyTorch abs()用法及代码示例

PyTorch torch.abs()方法计算给定输入张量的逐元素绝对值。

用法： torch.abs(inp, out=None) ? Tensor

参数

inp:这是输入张量。
out:这是可选参数，是输出张量。

返回：它返回具有输入inp绝对值的张量。

让我们借助几个示例来了解这个概念：
范例1：

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size 1  
a = torch.FloatTensor([-15])  
print(a)  
    
# Applying the abs function and  
# storing the result in 'b'  
b = torch.abs(a)  
print(b)

输出：

-15
[torch.FloatTensor of size 1]
 15
[torch.FloatTensor of size 1]

范例2：

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size n  
a = torch.FloatTensor([15, -5, 3, -2])  
print(a)  
    
# Applying the abs function and  
# storing the result in 'b'  
b = torch.abs(a)  
print(b)

输出：

 15
 -5
  3
 -2
[torch.FloatTensor of size 4]
 15
  5
  3
  2
[torch.FloatTensor of size 4]

相关用法

注：本文由纯净天空筛选整理自PranchalKatiyar大神的英文原创作品 Python – PyTorch abs() method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。