Fonction torch.softmax de PyTorch
Manuel de référence Pytorch torch
torch.softmaxest une fonction de PyTorch utilisée pour calculer la fonction softmax élément par élément.
Définition de fonction
torch.softmax(input, dim, dtype=None)
Exemple d'utilisation
Exemple
import torch
# Calculer le softmax
x = torch.tensor([1.0, 2.0, 3.0])
result = torch.softmax(x, dim=0)
print(result)
# Calculer le softmax
x = torch.tensor([1.0, 2.0, 3.0])
result = torch.softmax(x, dim=0)
print(result)
Autres extensions