Skip to content

GitLab

Unverified Commit a2defae2 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

[Bug] Fix typo in `_gsddmm_hetero` (#4944)

parent 84a12cf6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/dgl/sparse.py
View file @ a2defae2
...@@ -605,7 +605,7 @@ def _gsddmm_hetero( ...@@ -605,7 +605,7 @@ def _gsddmm_hetero(
expand_lhs = True expand_lhs = True
if use_rhs: if use_rhs:
if rhs is not None and F.ndim(rhs) == 1: if rhs is not None and F.ndim(rhs) == 1:
rhs = F.unsqueeze(lhs, -1) rhs = F.unsqueeze(rhs, -1)
expand_rhs = True expand_rhs = True
ctx = F.context(lhs) if use_lhs else F.context(rhs) ctx = F.context(lhs) if use_lhs else F.context(rhs)
dtype = F.dtype(lhs) if use_lhs else F.dtype(rhs) dtype = F.dtype(lhs) if use_lhs else F.dtype(rhs)
......
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