jax.experimental.sparse.bcsr_extract# jax.experimental.sparse.bcsr_extract(indices, indptr, mat)[source]# Extract values from a dense matrix at given BCSR (indices, indptr). Parameters: indices (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – An ndarray; see BCSR indices. indptr (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – An ndarray; see BCSR indptr. mat (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – A dense matrix. Return type: Array Returns: An ndarray; see BCSR data.