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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。