jax.scipy.linalg.pascal

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

创建 n 阶的 Pascal 矩阵近似。

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

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

参数:

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

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

返回:

形状为 (n, n) 的 Pascal 矩阵

返回类型:

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.]]