Fonction torch.sgn de PyTorch

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

torch.sgnest une fonction de PyTorch utilisée pour retourner le vecteur de signe élément par élément. Pour les nombres complexes, elle retourne le nombre complexe unitaire.

Définition de fonction

torch.sgn(input, out=None)

Exemple d'utilisation

Exemple

import torch

# Retourner le vecteur de signe élément par élément
x = torch.tensor([-2.0, 0.0, 3.0])
result = torch.sgn(x)
print(result)

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

Autres extensions