Skip to content

GitLab

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

32 (#5216) 


Co-authored-by: default avatarSteve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
parent 17fad103
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
python/dgl/sparse/diag_matrix.py
View file @ 7a2dd5ae
......@@ -237,8 +237,8 @@ class DiagMatrix:
return self.to(device="cpu")
def float(self):
"""Converts the matrix values to float data type. If the matrix already
uses float data type, the original matrix will be returned.
"""Converts the matrix values to float32 data type. If the matrix
already uses float data type, the original matrix will be returned.
Returns
-------
......@@ -277,7 +277,7 @@ class DiagMatrix:
return self.to(dtype=torch.double)
def int(self):
"""Converts the matrix values to int data type. If the matrix already
"""Converts the matrix values to int32 data type. If the matrix already
uses int data type, the original matrix will be returned.
Returns
......
python/dgl/sparse/sparse_matrix.py
View file @ 7a2dd5ae
......@@ -324,8 +324,8 @@ class SparseMatrix:
return self.to(device="cpu")
def float(self):
"""Converts the matrix values to float data type. If the matrix already
uses float data type, the original matrix will be returned.
"""Converts the matrix values to float32 data type. If the matrix
already uses float data type, the original matrix will be returned.
Returns
-------
......@@ -371,7 +371,7 @@ class SparseMatrix:
return self.to(dtype=torch.double)
def int(self):
"""Converts the matrix values to int data type. If the matrix already
"""Converts the matrix values to int32 data type. If the matrix already
uses int data type, the original matrix will be returned.
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