Fonction torch.greater de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.greaterIl s'agit d'une fonction de PyTorch pour la comparaison élément par élément supérieure à. Renvoie un tenseur avec True aux positions où les éléments du premier tenseur sont supérieurs aux éléments correspondants du second tenseur.

Définition de fonction

torch.greater(input, other)

Exemple d'utilisation

Exemple

import torch

# Créer deux tenseurs
x = torch.tensor([3, 5, 2])
y = torch.tensor([2, 4, 2])

# Comparer élément par élément supérieur à
result = torch.greater(x, y)
print(result)

Le résultat de sortie est:

tensor([True, True, False])

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions