jax.scipy.linalg.pascal#

jax.scipy.linalg.pascal(n, kind=None)[源代码]#

创建 n 阶 Pascal 矩阵近似。

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

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