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。

返回:

包含阶乘值的数组。

返回类型:

数组

注意事项

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

另请参阅

jax.scipy.special.gamma()