Función torch.greater de PyTorch
Manual de referencia de PyTorch torch
torch.greaterEs una función en PyTorch para comparar mayor que elemento por elemento. Devuelve un tensor donde la posición es True si el elemento del primer tensor es mayor que el elemento correspondiente del segundo tensor.
Definición de la función
torch.greater(input, other)
Ejemplo de uso
Ejemplo
import torch
# Crear dos tensores
x = torch.tensor([3, 5, 2])
y = torch.tensor([2, 4, 2])
# Comparar elemento por elemento si es mayor
result = torch.greater(x, y)
print(result)
# Crear dos tensores
x = torch.tensor([3, 5, 2])
y = torch.tensor([2, 4, 2])
# Comparar elemento por elemento si es mayor
result = torch.greater(x, y)
print(result)
El resultado de salida es:
tensor([True, True, False])
Otras extensiones