event_csrmm

Contents

event_csrmm#

class braintaichi.event_csrmm(data, indices, indptr, matrix, *, shape, transpose=False)[source]#

Product of CSR sparse matrix and a dense event matrix.

Parameters:

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

  • 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

  • matrix (Union[Array, ndarray, bool, number, bool, int, float, complex]) – array of shape (shape[0] if transpose else shape[1], cols) and dtype data.dtype

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

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

Returns:

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

Return type:

C

Contents