search

cszhangzhen / HGP-SL

Hierarchical Graph Pooling with Structure Learning
325 stars 54 forks source link

About the GAT model used in this paper #16

Open Riofd opened 2 years ago

Riofd commented 2 years ago

Hello, thank you for your great contributions! I want to know how to deal with the dimensional problem when I use the GAT model with multi-head, where x3 is with the average feature, while x1 and x2 are with the concatenation feature. x = F.relu(x1) + F.relu(x2) + F.relu(x3) this would make a dimensional error. Thanks!

cszhangzhen commented 2 years ago

Hi, actually you can use average feature in x1,x2,x3 to make it work.