jax.numpy.linalg.matrix_power#

jax.numpy.linalg.matrix_power(a, n)[原始碼]#

將方陣提升為整數次方。

JAX 實現的 numpy.linalg.matrix_power()，透過重複平方實現。

參數:

  • a (ArrayLike) – 形狀為 (..., M, M) 的陣列，將被提升到次方 n。

  • n (int) – 矩陣應提升到的整數指數。

返回:

形狀為 (..., M, M) 的陣列，包含 a 的 n 次方矩陣。

返回類型:

Array

範例

>>> a = jnp.array([[1., 2.],
...                [3., 4.]])
>>> jnp.linalg.matrix_power(a, 3)
Array([[ 37.,  54.],
       [ 81., 118.]], dtype=float32)
>>> a @ a @ a  # equivalent evaluated directly
Array([[ 37.,  54.],
       [ 81., 118.]], dtype=float32)

這也支援零次方

>>> jnp.linalg.matrix_power(a, 0)
Array([[1., 0.],
       [0., 1.]], dtype=float32)

也支援負次方

>>> with jnp.printoptions(precision=3):
...   jnp.linalg.matrix_power(a, -2)
Array([[ 5.5 , -2.5 ],
       [-3.75,  1.75]], dtype=float32)

負次方等同於逆矩陣的矩陣乘法

>>> inv_a = jnp.linalg.inv(a)
>>> with jnp.printoptions(precision=3):
...   inv_a @ inv_a
Array([[ 5.5 , -2.5 ],
       [-3.75,  1.75]], dtype=float32)