Pytorch implementation of the Graph Attention Network model by Veličković et. al (2017, https://arxiv.org/abs/1710.10903)
2.89k
stars
689
forks
source link
Make _prepate_attentional_mechanism_input a bit more readable #44
Closed
ahmad-PH closed 4 years ago
Replaced .repeat(...).view(...) with .repeat_interleave(...) which does the exact same thing in less code.
@Diego999