torch_i0 function

I0

I0

torch_i0(self)

Arguments

  • self: (Tensor) the input tensor

i0(input, *, out=None) -> Tensor

Computes the zeroth order modified Bessel function of the first kind for each element of input.

\mboxouti=I0(\mboxinputi)=k=0(\mboxinputi2/4)k(k!)2 \mbox{out}_{i} = I_0(\mbox{input}_{i}) = \sum_{k=0}^{\infty} \frac{(\mbox{input}_{i}^2/4)^k}{(k!)^2}

Examples

if (torch_is_installed()) { torch_i0(torch_arange(start = 0, end = 5, dtype=torch_float32())) }
  • Maintainer: Daniel Falbel
  • License: MIT + file LICENSE
  • Last published: 2025-02-14