src.sparse_torch.csx_matrixΒΆ
Functions
Concatenate multiple sparse matrix in first (batch) dimension. |
|
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