jax.experimental.sparse.coo_matmat# jax.experimental.sparse.coo_matmat(mat, B, *, transpose=False)[source]# Product of COO sparse matrix and a dense matrix. Parameters: mat (COO) – COO matrix B (Array) – array of shape (mat.shape[0] if transpose else mat.shape[1], cols) and dtype mat.dtype transpose (bool) – boolean specifying whether to transpose the sparse matrix before computing. Returns: array of shape (mat.shape[1] if transpose else mat.shape[0], cols)representing the matrix vector product. Return type: C
