csrmm

Contents

csrmm#

class braintaichi.csrmm(data, indices, indptr, matrix, *, shape, transpose=False)#

Product of CSR sparse matrix and a dense matrix.

Parameters:

  • data (Union[Array, ndarray, bool, number, bool, int, float, complex, Quantity]) – array of shape (nse,).

  • indices (Union[Array, ndarray, bool, number, bool, int, float, complex]) – array of shape (nse,)

  • indptr (Union[Array, ndarray, bool, number, bool, int, float, complex]) – array of shape (shape[0] + 1,) and dtype indices.dtype

  • B – array of shape (shape[0] if transpose else shape[1], cols) and

  • data.dtype (dtype)

  • shape (Tuple[int, int]) – length-2 tuple representing the matrix shape

  • transpose (bool) – boolean specifying whether to transpose the sparse matrix

  • computing. (before)

Returns:

array of shape (shape[1] if transpose else shape[0], cols) representing the matrix-matrix product.

Return type:

C

Contents