jax.scipy.special.factorial

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

阶乘函数

JAX实现的 scipy.special.factorial

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

参数:

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

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

返回:

包含阶乘值的数组。

返回类型:

数组

注释

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

另请参阅

jax.scipy.special.gamma()