Fonction torch.greater_equal de PyTorch
Manuel de référence Pytorch torch
torch.greater_equalIl s'agit d'une fonction de PyTorch pour la comparaison élément par élément de supérieur ou égal. Elle renvoie un tenseur avec True là où chaque élément du premier tenseur est supérieur ou égal à l'élément correspondant du second tenseur.
Définition de fonction
torch.greater_equal(input, other)
Exemple d'utilisation
Exemple
import torch
# Créer deux tenseurs
x = torch.tensor([3, 5, 2])
y = torch.tensor([2, 4, 2])
# Comparaison élément par élément supérieur ou égal
result = torch.greater_equal(x, y)
print(result)
# Créer deux tenseurs
x = torch.tensor([3, 5, 2])
y = torch.tensor([2, 4, 2])
# Comparaison élément par élément supérieur ou égal
result = torch.greater_equal(x, y)
print(result)
La sortie est :
tensor([True, True, True])
Autres extensions