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])