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 (ArrayLike) – 类似数组，将逐元素计算阶乘的值

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

返回:

包含阶乘值的数组。

返回类型:

Array

注释

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

另请参阅

jax.scipy.special.gamma()