mindspore.ops.BesselK0

查看源文件
class mindspore.ops.BesselK0[源代码]

逐元素计算第二类零阶修正Bessel函数值。

计算公式定义如下:

\[\begin{split}\begin{array}{ll} \\ K_{0}(x)= \lim_{\nu \to 0} \left(\frac{\pi}{2}\right) \frac {I_{-\nu}(x)-I_{\nu}(x)}{\sin (\nu \pi)} = \int_{0}^{\infty} e^{-x \cosh t} d t \end{array}\end{split}\]

其中 \(I_{0}\) 是第一类零阶修正Bessel函数。

警告

这是一个实验性API,后续可能修改或删除。

输入:
  • x (Tensor) - 输入Tensor。数据类型应为float16、float32或float64。

输出:

Tensor,shape和数据类型与 x 相同。

异常:
  • TypeError - x 不是float16、float32或float64数据类型的Tensor。

支持平台:

GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> bessel_k0 = ops.BesselK0()
>>> x = Tensor(np.array([0.24, 0.83, 0.31, 0.09]), mindspore.float32)
>>> output = bessel_k0(x)
>>> print(output)
[1.579826  0.5402144 1.3424659 2.5310173]