JAX study notes[11]

the maxtrix manipulation

the following matrix will be demonstrated for manipulation.

a = jnp.array([[1, 2], [3, 4]])
b = jnp.array([[5, 6], [7, 8]])

1. matrix addition

sum = a + b

2. matrix multiplication

elementwise = a * b

3. Matrix dot product

dot = jnp.dot(a, b)
dot = a @ b

4. matrix transposition

transpose = a.T

5. inverse matrix

inv = jnp.linalg.inv(a)

6. determinant

det = jnp.linalg.det(a)

references

1. deepseek