Python PyTorch i0e用法及代碼示例

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

用法:

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

參數：

input(Tensor) -輸入張量。

關鍵字參數：

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

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

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

例子：

>>> torch.special.i0e(torch.arange(5, dtype=torch.float32))
tensor([1.0000, 0.4658, 0.3085, 0.2430, 0.2070])