models.nn.gat
¶
Module Contents¶
Classes¶
Simple GAT layer, similar to https://arxiv.org/abs/1710.10903 |
|
Special function for only sparse region backpropataion layer. |
|
Sparse version GAT layer, similar to https://arxiv.org/abs/1710.10903 |
|
-
class
models.nn.gat.
GraphAttentionLayer
(in_features, out_features, dropout, alpha, concat=True)[source]¶ Bases:
torch.nn.Module
Simple GAT layer, similar to https://arxiv.org/abs/1710.10903
-
class
models.nn.gat.
SpecialSpmmFunction
[source]¶ Bases:
torch.autograd.Function
Special function for only sparse region backpropataion layer.
-
class
models.nn.gat.
SpGraphAttentionLayer
(in_features, out_features, dropout, alpha, concat=True)[source]¶ Bases:
torch.nn.Module
Sparse version GAT layer, similar to https://arxiv.org/abs/1710.10903
-
class
models.nn.gat.
PetarVGAT
(nfeat, nhid, nclass, dropout, alpha, nheads)[source]¶ Bases:
models.BaseModel