Python – Método PyTorch add()

El método PyTorch torch.add()agrega un valor constante a cada elemento del tensor de entrada y devuelve un nuevo tensor modificado.

Sintaxis: torch.add(inp, c, out=None)

Argumentos

• inp: Este es el tensor de entrada.
• c: El valor que se le va a sumar a cada elemento del tensor.
• out: Este es un parámetro opcional y es el tensor de salida.

Return: Devuelve un Tensor.

Veamos este concepto con la ayuda de algunos ejemplos:
Ejemplo 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)

Producción:

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]

Ejemplo 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) 

Producción:

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