PyTorch torch.logsumexp 関数
Pytorch torch リファレンスマニュアルtorch.logsumexpPyTorchでテンソル要素の指数の対数の和を計算するための関数です。
関数定義
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])
その他の拡張