PyTorch torch.logsumexp 函数
Pytorch torch 参考手册torch.logsumexp 是 PyTorch 中用于计算张量元素指数的对数和的函数。
函数定义
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])