src.sparse_torch.csx_matrixΒΆ

Functions

concatenate

Concatenate multiple sparse matrix in first (batch) dimension.

dot

Compute the dot product of one CSR/dense matrix and one CSC matrix. The result will be returned in a new CSR or dense matrix. Note that only a few combinations of types are implemented. :param t1: fist input matrix :param t2: second input matrix :param dense_output: output matrix in dense format :return: dot result in new csr matrix (dense=False) or dot result in dense matrix (dense=True).

Classes

CSCMatrix3d

CSRMatrix3d

CSXMatrix3d