PyTorch torch.logsumexp Funktion
PyTorch torch-Referenzhandbuch
torch.logsumexpEs ist eine Funktion in PyTorch, die zur Berechnung der Log-Sum-Exp von Tensor-Elementen verwendet wird.
Funktionsdefinition
torch.logsumexp(input, dim, keepdim=False, out=None)
Anwendungsbeispiel
Beispiel
import torch
# Tensor erstellen
x = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
# Berechne logsumexp nach Dimension
y = torch.logsumexp(x, dim=1)
print("logsumexp:", y)
# Tensor erstellen
x = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
# Berechne logsumexp nach Dimension
y = torch.logsumexp(x, dim=1)
print("logsumexp:", y)
Das Ausgabeergebnis ist:
logsumexp: tensor([ 3.4076, 6.4076])
Andere Erweiterungen