cr.nimble.off_diagonal_elements

cr.nimble.off_diagonal_elements(A)

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

(jax.numpy.ndarray)