jax.scipy.special.factorial#
- jax.scipy.special.factorial(n, exact=False)[source]#
階乘函數
JAX 實作的
scipy.special.factorial\[\mathrm{factorial}(n) = n! = \prod_{k=1}^n k\]- 參數:
n (ArrayLike) – 類陣列, 將逐元素計算階乘的值
exact (bool) – bool, 僅支援
exact=False。
- 回傳值:
包含階乘值的陣列。
- 回傳類型:
筆記
這會透過
gamma()函數計算浮點數值的階乘。JAX 不支援精確階乘，因為它不是特別有用：當n=20以上時，精確結果無法以 64 位元整數表示，而 64 位元整數是 JAX 可用的最大整數。