Función torch.minimum de PyTorch
Manual de referencia de Pytorch torch
torch.minimumEs una función utilizada para tomar el mínimo elemento por elemento en PyTorch. Devuelve un nuevo tensor compuesto por los valores mínimos de los elementos correspondientes de dos tensores.
Definición de la función
torch.minimum(input, other)
Ejemplo de uso
Ejemplo
import torch
# Crear dos tensores
x = torch.tensor([3, 1, 4])
y = torch.tensor([1, 5, 2])
# Tomar el mínimo elemento por elemento
result = torch.minimum(x, y)
print(result)
# Crear dos tensores
x = torch.tensor([3, 1, 4])
y = torch.tensor([1, 5, 2])
# Tomar el mínimo elemento por elemento
result = torch.minimum(x, y)
print(result)
El resultado de salida es:
tensor([1, 1, 2])
Otras extensiones