Python PyTorch expm1用法及代碼示例

本文簡要介紹python語言中 torch.special.expm1 的用法。

用法:

torch.special.expm1(input, *, out=None) → Tensor

參數：

input(Tensor) -輸入張量。

關鍵字參數：

out(Tensor,可選的) -輸出張量。

計算 input 的元素減 1 的 index 。

y_{i} = e^{x_{i}} - 1

注意

對於較小的 x 值，此函數提供比 exp(x) - 1 更高的精度。

例子：

>>> torch.special.expm1(torch.tensor([0, math.log(2.)]))
tensor([ 0.,  1.])