Python PyTorch i0用法及代碼示例

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

用法:

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

參數：

input(Tensor) -輸入張量

關鍵字參數：

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

計算 input 的每個元素的第一類零階修正貝塞爾函數。

\text{out}_{i} = I_0(\text{input}_{i}) = \sum_{k=0}^{\infty} \frac{(\text{input}_{i}^2/4)^k}{(k!)^2}

例子：

>>> torch.i0(torch.arange(5, dtype=torch.float32))
tensor([ 1.0000,  1.2661,  2.2796,  4.8808, 11.3019])