PyTorch torch.logaddexp 関数
Pytorch torch リファレンスマニュアルtorch.logaddexpは、PyTorch で log(exp(x) + exp(y)) を計算するための関数です。
関数定義
torch.logaddexp(input, other)
使用例
実例
import torch
# log(exp(x) + exp(y)) を計算
x = torch.tensor([1.0, 2.0])
y = torch.tensor([3.0, 4.0])
z = torch.logaddexp(x, y)
print("logaddexp:", z)
# log(exp(x) + exp(y)) を計算
x = torch.tensor([1.0, 2.0])
y = torch.tensor([3.0, 4.0])
z = torch.logaddexp(x, y)
print("logaddexp:", z)
出力結果は次のとおりです。
logaddexp: tensor([3.0486, 4.0181])
その他の拡張