Fonction torch.sigmoid de PyTorch
Manuel de référence Pytorch torch
torch.sigmoidC'est la fonction dans PyTorch utilisée pour calculer la fonction sigmoïde élément par élément. La fonction sigmoïde est définie comme 1/(1+exp(-x)).
Définition de fonction
torch.sigmoid(input, out=None)
Exemple d'utilisation
Exemple
import torch
# Calculer la fonction sigmoïde
x = torch.tensor([0.0, 1.0, -1.0])
result = torch.sigmoid(x)
print(result)
# Calculer la fonction sigmoïde
x = torch.tensor([0.0, 1.0, -1.0])
result = torch.sigmoid(x)
print(result)
Autres extensions