Python PyTorch add()用法及代碼示例

PyTorch torch.add()方法將常數值添加到輸入張量的每個元素，並返回新的修改後的張量。

用法： torch.add(inp, c, out=None)

參數

inp:這是輸入張量。
c:要添加到張量的每個元素的值。
out:這是可選參數，它是輸出張量。

返回：它返回張量。

讓我們借助幾個示例來了解這個概念：
範例1：

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size 6 
a = torch.randn(6)  
print(a)  
    
# Applying the add function and  
# storing the result in 'b'  
b = torch.add(a, 5)  
print(b)

輸出：

0.2403
 1.3826
-0.1763
-1.5177
-0.0555
 1.4558
[torch.FloatTensor of size 6]
 5.2403
 6.3826
 4.8237
 3.4823
 4.9445
 6.4558
[torch.FloatTensor of size 6]

範例2：

# Importing the PyTorch library  
import torch  
    
# A constant tensor of size 6 
a = torch.FloatTensor([1, 3, 8, 4, 10])  
print(a)  
    
# Applying the add function and  
# storing the result in 'b'  
b = torch.add(a, 5)  
print(b)

輸出：

 1
  3
  8
  4
 10
[torch.FloatTensor of size 5]
  6
  8
 13
  9
 15
[torch.FloatTensor of size 5]

相關用法

注：本文由純淨天空篩選整理自PranchalKatiyar大神的英文原創作品 Python – PyTorch add() method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。