PyTorch torch.logsumexp 函数

PyTorch torch.logsumexp 函数 Pytorch torch 参考手册 torch.logsumexp 是 PyTorch 中用于计算张量元素指数的对数和的函数。函数定义 torch.logsumexp(input, dim, keepdim=False, out=None) 使用示例实例 [mycode4 type='python'] import torch # 创建..

更新于 2026/7/16

教程详情页默认 noindex。

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)

输出结果为：

logsumexp: tensor([ 3.4076, 6.4076])

Pytorch torch 参考手册