jax.scipy.linalg.pascal

jax.scipy.linalg.pascal(n, kind=None)

创建 n 阶帕斯卡矩阵近似。

scipy.linalg.pascal() 的 JAX 实现。

帕斯卡矩阵的元素近似于二项式系数。此实现并不精确，因为 JAX 不支持精确阶乘。

参数:

• n (int) – 要创建的矩阵的大小。

• kind (str | None) – （可选）必须是 lower、upper 或 symmetric（默认）之一。

返回:

形状为 (n, n) 的帕斯卡矩阵

返回类型:

Array

示例

>>> with jnp.printoptions(precision=3):
...   print(jax.scipy.linalg.pascal(3, kind="lower"))
...   print(jax.scipy.linalg.pascal(4, kind="upper"))
...   print(jax.scipy.linalg.pascal(5))
[[1. 0. 0.]
 [1. 1. 0.]
 [1. 2. 1.]]
[[1. 1. 1. 1.]
 [0. 1. 2. 3.]
 [0. 0. 1. 3.]
 [0. 0. 0. 1.]]
[[ 1.  1.  1.  1.  1.]
 [ 1.  2.  3.  4.  5.]
 [ 1.  3.  6. 10. 15.]
 [ 1.  4. 10. 20. 35.]
 [ 1.  5. 15. 35. 70.]]