快捷方式

torch.logaddexp

torch.logaddexp(input, other, *, out=None) Tensor

输入的指数运算之和的对数。

逐点计算 log(ex+ey)\log\left(e^x + e^y\right)。此函数在统计学中很有用,在统计学中,计算出的事件概率可能非常小,以至于超出普通浮点数的范围。在这种情况下,将存储计算出的概率的对数。此函数允许以这种方式存储的概率相加。

此操作应与 torch.logsumexp() 区分开来,后者对单个张量执行归约。

参数
  • input (张量) – 输入张量。

  • other (张量) – 第二个输入张量

关键字参数

out (张量, 可选) – 输出张量。

示例

>>> torch.logaddexp(torch.tensor([-1.0]), torch.tensor([-1.0, -2, -3]))
tensor([-0.3069, -0.6867, -0.8731])
>>> torch.logaddexp(torch.tensor([-100.0, -200, -300]), torch.tensor([-1.0, -2, -3]))
tensor([-1., -2., -3.])
>>> torch.logaddexp(torch.tensor([1.0, 2000, 30000]), torch.tensor([-1.0, -2, -3]))
tensor([1.1269e+00, 2.0000e+03, 3.0000e+04])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源