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:
- Returns:
An ndarray; see BCSR data.