Skip to content

GitLab

Unverified Commit b743f767 authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

fix (#5151) 


Co-authored-by: default avatarSteve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
parent f94fa9d9
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
python/dgl/sparse/sparse_matrix.py
View file @ b743f767
...@@ -103,7 +103,8 @@ class SparseMatrix: ...@@ -103,7 +103,8 @@ class SparseMatrix:
return self.c_sparse_matrix.coo() return self.c_sparse_matrix.coo()
def csr(self) -> Tuple[torch.Tensor, ...]: def csr(self) -> Tuple[torch.Tensor, ...]:
"""Get the coordinate (COO) representation of the sparse matrix. r"""Get the compressed sparse row (CSR) representation of the sparse
matrix.
Returns Returns
------- -------
...@@ -114,7 +115,8 @@ class SparseMatrix: ...@@ -114,7 +115,8 @@ class SparseMatrix:
return self.c_sparse_matrix.csr() return self.c_sparse_matrix.csr()
def csc(self) -> Tuple[torch.Tensor, ...]: def csc(self) -> Tuple[torch.Tensor, ...]:
"""Get the coordinate (COO) representation of the sparse matrix. r"""Get the compressed sparse column (CSC) representation of the sparse
matrix.
Returns Returns
------- -------
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment