Fonction torch.sign de PyTorch

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

torch.signC'est une fonction dans PyTorch utilisée pour renvoyer le signe élément par élément. Elle renvoie -1 (négatif), 0 (zéro), ou 1 (positif).

Définition de fonction

torch.sign(input, out=None)

Exemples d'utilisation

Exemple

import torch

# Renvoyer le signe élément par élément
x = torch.tensor([-2.0, 0.0, 3.0])
result = torch.sign(x)
print(result)

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

Autres extensions