¿Cómo realizar una resta de elementos en tensores en PyTorch?

En este artículo, vamos a entender cómo realizar una resta por elementos en tensores en PyTorch en Python. Podemos realizar una resta por elementos usando el método torch.sub().

El método torch.sub() nos permite realizar restas en las mismas o diferentes dimensiones de tensores. Toma dos tensores como entradas y devuelve un nuevo tensor con el resultado (resta por elementos). Si los tensores tienen diferentes dimensiones, devolverá el tensor de mayor dimensión. también podemos restar una cantidad escalar con un tensor usando la función torch.sub(). Podemos usar la siguiente sintaxis para calcular la resta por elementos.

Sintaxis: torch.sub(entrada, otro, *, alfa=1, salida=Ninguno)

Parámetros:

• entrada: el tensor de entrada.
• otro: Este es el tensor o número a restar del tensor de entrada.
• alfa (Número): el parámetro es multiplicador de otro.
• out: es el tensor de salida, este es un parámetro opcional.

Retorno: devolverá un nuevo tensor modificado con la resta por elementos del tensor ingresado por el otro tensor.

Ejemplo 1:

El siguiente programa es para realizar una resta por elementos 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(" First Tensor: ", tens_1)
print(" Second Tensor: ", tens_2)
 
# multiply tensors
tens = torch.sub(tens_1, tens_2)
 
# display result after perform element wise
# subtraction
print(" After Element-wise subtraction: ", tens)

Salida :

Ejemplo 2:

El siguiente programa es para realizar una resta por elementos en dos tensores 2D.

# import torch library
import torch
 
# define two tensors
tens_1 = torch.Tensor([[10, 20], [30, 40]])
 
tens_2 = torch.Tensor([[1, 2], [3, 4]])
 
# display tensors
print("First Tensor:", tens_1)
print("Second Tensor:", tens_2)
 
# Subtract tensors
tens = torch.sub(tens_1, tens_2)
 
# display result after perform element wise
# subtraction
print("After Element-wise subtraction:", tens)

Producción:

Ejemplo 3:

El siguiente programa es para saber restar una cantidad escalar a un tensor.

# import torch library
import torch
 
# define a tensor
tens = torch.Tensor([100, 200, 300, 400, 500])
 
# display tensors
print(tens)
 
# Subtract 50 from tensor
tens_result = torch.sub(tens, 50)
 
# display result after subtract scalar from tensor
print("After subtract scalar from tensor:", tens_result)

Producción:

Ejemplo 4:

El siguiente programa es para comprender cómo realizar una resta por elementos en dos tensores de dimensión diferentes.

# import torch library
import torch
 
# define 2D tensor
tens_1 = torch.Tensor([[100, 200], [300, 400]])
 
# define 1D tensor
tens_2 = torch.Tensor([10, 20])
 
# display tensors
print("First Tensor:", tens_1)
print("Second Tensor:", tens_2)
 
# Subtract tensors
tens = torch.sub(tens_1, tens_2)
 
# display result after perform element wise subtraction
print("After Element-wise subtraction:", tens)

Producción: