PyTorch torch.greater_equal función
Pytorch torch manual de referencia
torch.greater_equalEs una función utilizada en PyTorch para comparar elemento a elemento si es mayor o igual. Devuelve un tensor en el que la posición es True si el elemento del primer tensor es mayor o igual que el elemento correspondiente del segundo tensor.
Definición de la función
torch.greater_equal(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 a elemento si es mayor o igual
result = torch.greater_equal(x, y)
print(result)
# Crear dos tensores
x = torch.tensor([3, 5, 2])
y = torch.tensor([2, 4, 2])
# Comparar elemento a elemento si es mayor o igual
result = torch.greater_equal(x, y)
print(result)
El resultado de salida es:
tensor([True, True, True])
Otras extensiones