PyTorch torch.softmax-Funktion
PyTorch torch Referenzhandbuch
torch.softmaxist eine Funktion in PyTorch, die zur elementweisen Berechnung der Softmax-Funktion verwendet wird.
Funktionsdefinition
torch.softmax(input, dim, dtype=None)
Anwendungsbeispiel
Beispiel
import torch
# Softmax berechnen
x = torch.tensor([1.0, 2.0, 3.0])
result = torch.softmax(x, dim=0)
print(result)
# Softmax berechnen
x = torch.tensor([1.0, 2.0, 3.0])
result = torch.softmax(x, dim=0)
print(result)
Andere Erweiterungen