Función torch.logsumexp de PyTorch
Manual de referencia de Pytorch torch
torch.logsumexpEs una función en PyTorch que se utiliza para calcular el logaritmo de la suma de exponenciales de los elementos de un tensor.
Definición de la función
torch.logsumexp(input, dim, keepdim=False, out=None)
Ejemplo de uso
Ejemplo
import torch
# Crear tensor
x = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
# Calcular logsumexp por dimensión
y = torch.logsumexp(x, dim=1)
print("logsumexp:", y)
# Crear tensor
x = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
# Calcular logsumexp por dimensión
y = torch.logsumexp(x, dim=1)
print("logsumexp:", y)
El resultado de salida es:
logsumexp: tensor([ 3.4076, 6.4076])
Otras extensiones