Python PyTorch i1用法及代码示例

本文简要介绍python语言中 torch.special.i1 的用法。

用法:

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

参数：

input(Tensor) -输入张量。

关键字参数：

out(Tensor,可选的) -输出张量。

计算 input 的每个元素的第一类修正贝塞尔函数(定义如下)。

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

例子：

>>> torch.special.i1(torch.arange(5, dtype=torch.float32))
tensor([0.0000, 0.5652, 1.5906, 3.9534, 9.7595])