Sparse matrix multiplication in pytorch