¿Cómo realizar la suma de elementos en tensores en PyTorch?

En este artículo, veremos cómo realizar sumas de elementos en tensores en PyTorch en Python. Podemos realizar sumas por elementos usando la función torch.add() . 

Esta función también nos permite realizar sumas en las mismas o diferentes dimensiones de tensores. Si los tensores tienen diferentes dimensiones, devolverá el tensor de mayor dimensión. 

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

Parámetros:

• 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 nuevo tensor modificado.

Ejemplo 1: El siguiente programa es para realizar sumas en dos tensores de una sola dimensión.

# import torch library
import torch
  
# define two tensors
tens_1 = torch.Tensor([10, 20, 30, 40, 50])
tens_2 = torch.Tensor([1, 2, 3, 4, 5])
  
# display tensors
print("\nFirst Tensor \n", tens_1)
print("\nsecond Tensor \n", tens_2)
  
# Add both the tensor
tens = torch.add(tens_1, tens_2)
print("\n After Element-wise Addition\n", tens)

Producción:

Ejemplo 2: El siguiente programa es para realizar la suma de elementos en tensores 2D.

# import the torch library
import torch
  
# define 2D tensors
tens_1 = torch.Tensor([[1, 2], [3, 4]])
tens_2 = torch.Tensor([[10, 20], [20, 40]])
  
# display these tensors
print("First Tensor \n", tens_1)
print("second Tensor \n", tens_2)
  
# perform addition on these tensors
tens = torch.add(tens_1, tens_2)
  
# display final result
print("\n After Element-wise Addition\n", tens)

Producción:

Ejemplo 3: El siguiente programa muestra cómo realizar la suma de elementos en dos tensores de dimensión diferentes.

# import torch library
import torch
  
# define 1D tensor
tens_1 = torch.Tensor([1, 2])
  
# define 2D tensor
tens_2 = torch.Tensor([[10, 20], [20, 40]])
  
# display tensors
print("\nFirst Tensor \n", tens_1)
print("\nsecond Tensor \n", tens_2)
  
# perform addition on tensors
tens = torch.add(tens_1, tens_2)
  
# display final output
print("\n After Element-wise Addition\n", tens)

Producción:

Ejemplo 4: El siguiente programa es para saber cómo sumar una cantidad escalar a un tensor.

# Python program to perform element-wise Addition
# import the required library
import torch
  
# define a tensor
tens_1 = torch.Tensor([1, 2, 3, 4])
  
# define 2D tensor
tens_2 = torch.Tensor([[10, 20], [30, 40]])
  
# display tensors
print("\nFirst Tensor \n", tens_1)
print("second Tensor \n", tens_2)
  
# perform addition on 1D tensors
t1 = torch.add(tens_1, 10)
# display final output
print("\n After adding scalar quantity to 1D tensor: \n", t1)
  
# perform addition on 2D tensors
t2 = torch.add(tens_2, 20)
# display final output
print("\n After adding scalar quantity to 2D tensor: \n", t2)

Producción: