jax.scipy.special.factorial

jax.scipy.special.factorial(n, exact=False)

阶乘函数

scipy.special.factorial 的 JAX 实现

\[\mathrm{factorial}(n) = n! = \prod_{k=1}^n k\]

参数:

• n (类数组) – 类数组，将按元素计算阶乘的值

• exact (bool) – 布尔值，仅支持 exact=False。

返回:

包含阶乘值的数组。

返回类型:

Array

注释

这个函数通过 gamma() 函数计算浮点数值的阶乘。JAX 不支持精确的阶乘，因为它不是特别有用：当 n=20 以上时，精确的结果无法用 64 位整数表示，而 64 位整数是 JAX 可用的最大整数。

另请参阅

jax.scipy.special.gamma()