Python PyTorch i1e用法及代碼示例

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

用法:

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

參數：

input(Tensor) -輸入張量。

關鍵字參數：

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

為 input 的每個元素計算第一類(定義如下)的 index 縮放的一階修正貝塞爾函數。

\text{out}_{i} = \exp(-|x|) * i1(x) = \exp(-|x|) * \frac{(\text{input}_{i})}{2} * \sum_{k=0}^{\infty} \frac{(\text{input}_{i}^2/4)^k}{(k!) * (k+1)!}

例子：

>>> torch.special.i1e(torch.arange(5, dtype=torch.float32))
tensor([0.0000, 0.2079, 0.2153, 0.1968, 0.1788])