mindspore.ops.BesselK1

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

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

计算公式定义如下:

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

其中 \(I_{1}\) 是第一类一阶修正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_k1 = ops.BesselK1()
>>> x = Tensor(np.array([0.24, 0.83, 0.31, 0.09]), mindspore.float32)
>>> output = bessel_k1(x)
>>> print(output)
[3.9190812  0.8143549  2.9440577 10.974864]