"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "2e83cbbb6de84be7241218c8f5ea914ceb68c149"
Unverified Commit 0ceb6751 authored by Maybewuss's avatar Maybewuss Committed by GitHub
Browse files

Correct error when using enisum (#2417)



* correct error when using enisum

* Correct error when using enisum
Co-authored-by: default avatarZihao Ye <expye@outlook.com>
parent 9dff5419
...@@ -73,7 +73,7 @@ class HGTLayer(nn.Module): ...@@ -73,7 +73,7 @@ class HGTLayer(nn.Module):
relation_pri = self.relation_pri[e_id] relation_pri = self.relation_pri[e_id]
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, relation_att)
v = torch.einsum("bij,ijk->bik", v, relation_msg) v = torch.einsum("bij,ijk->bik", v, relation_msg)
sub_graph.srcdata['k'] = k sub_graph.srcdata['k'] = k
......
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