Fonction torch.logsumexp de PyTorch
Manuel de référence PyTorch torch
torch.logsumexpC'est une fonction de PyTorch utilisée pour calculer le log-sum-exp des éléments du tenseur.
Définition de fonction
torch.logsumexp(input, dim, keepdim=False, out=None)
Exemple d'utilisation
Exemple
import torch
# Créer un tenseur
x = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
# Calculer logsumexp par dimension
y = torch.logsumexp(x, dim=1)
print("logsumexp:", y)
# Créer un tenseur
x = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
# Calculer logsumexp par dimension
y = torch.logsumexp(x, dim=1)
print("logsumexp:", y)
Le résultat de sortie est:
logsumexp: tensor([ 3.4076, 6.4076])
Autres extensions