src.sparse_torch.csx_matrix

Functions

concatenate	Concatenate multiple sparse matrix in first (batch) dimension.
dot	Compute the dot product of one CSR matrix and one CSC matrix. The result will be returned in a new CSR or dense matrix. Note that CUDA implementations do not work when dense=False. :param csr: fist input CSR matrix :param csc: second input CSC matrix :param dense: 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