cr.sparse.off_diagonal_elements¶
- cr.sparse.off_diagonal_elements(A)[source]¶
Returns the off diagonal elements of a matrix A
- Parameters
A (jax.numpy.ndarray) – A real 2D matrix
- Returns
A vector of off-diagonal elements in A
- Return type
Returns the off diagonal elements of a matrix A
A (jax.numpy.ndarray) – A real 2D matrix
A vector of off-diagonal elements in A