Fonction torch.gt de PyTorch


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

torch.gtest une fonction de PyTorch utilisée pour la comparaison élément par élément « supérieur à ».

Définition de fonction

torch.gt(input, other, out)

Exemple d'utilisation

Exemple

import torch

a = torch.tensor([1, 2, 3, 4])
b = torch.tensor([2, 2, 2, 2])

result = torch.gt(a, b)

print(result)

Le résultat de sortie est:

tensor([False, False,  True,  True])

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

Autres extensions