Функция torch.logsumexp в PyTorch
Справочное руководство PyTorch torch
torch.logsumexpЭто функция в PyTorch, используемая для вычисления log-sum-exp элементов тензора.
Определение функции
torch.logsumexp(input, dim, keepdim=False, out=None)
Пример использования
Пример
import torch
# Создать тензор
x = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
# Вычислить logsumexp по измерению
y = torch.logsumexp(x, dim=1)
print("logsumexp:", y)
# Создать тензор
x = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
# Вычислить logsumexp по измерению
y = torch.logsumexp(x, dim=1)
print("logsumexp:", y)
Результат вывода:
logsumexp: tensor([ 3.4076, 6.4076])
Другие расширения