Skip to content

GitLab

Unverified Commit ee8a0e66 authored by Maybewuss's avatar Maybewuss Committed by GitHub
Browse files

correct error when using enisum (#2406) 


Co-authored-by: default avatarZihao Ye <expye@outlook.com>
parent 1c976222
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/hgt/model.py
View file @ ee8a0e66
...@@ -74,7 +74,7 @@ class HGTLayer(nn.Module): ...@@ -74,7 +74,7 @@ class HGTLayer(nn.Module):
relation_msg = self.relation_msg[e_id] relation_msg = self.relation_msg[e_id]
k = torch.einsum("bij,ijk->bik", k, realtion_att) k = torch.einsum("bij,ijk->bik", k, realtion_att)
v = torch.einsum("bij,ijk->bik", k, relation_msg) v = torch.einsum("bij,ijk->bik", v, relation_msg)
sub_graph.srcdata['k'] = k sub_graph.srcdata['k'] = k
sub_graph.dstdata['q'] = q sub_graph.dstdata['q'] = q
......
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