Función torch.maximum de PyTorch
Manual de referencia de Pytorch torch
torch.maximumes una función en PyTorch para tomar el máximo elemento por elemento. Devuelve un nuevo tensor compuesto por los máximos de los elementos correspondientes de dos tensores.
Definición de la función
torch.maximum(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áximo elemento por elemento
result = torch.maximum(x, y)
print(result)
# Crear dos tensores
x = torch.tensor([3, 1, 4])
y = torch.tensor([1, 5, 2])
# Tomar el máximo elemento por elemento
result = torch.maximum(x, y)
print(result)
El resultado de salida es:
tensor([3, 5, 4])
Otras extensiones