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