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()