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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。