快捷方式

torch.logaddexp

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

輸入冪之和的對數。

逐點計算 log(ex+ey)\log\left(e^x + e^y\right)。此函式在統計學中非常有用,因為某些事件的計算機率可能非常小,以至於超出普通浮點數的範圍。在這種情況下,會儲存計算機率的對數。此函式允許對以這種方式儲存的機率進行相加。

應將此操作與對單個張量執行歸約的 torch.logsumexp() 函式區分開來。

引數
  • input (Tensor) – 輸入張量。

  • other (Tensor) – 第二個輸入張量

關鍵字引數

out (Tensor, optional) – 輸出張量。

示例

>>> 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 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源