pygod.nn.encoder¶
Graph Neural Networks Encoders
- class pygod.nn.encoder.GNA(in_channels, hidden_channels, num_layers, out_channels, dropout, act)[source]¶
Graph Node Attention Network (GNA). See [YZY+21] for more details.
- forward(s, edge_index)[source]¶
Forward computation.
- Parameters:
s (torch.Tensor) – Input node embeddings.
edge_index (torch.Tensor) – Edge index.
- Returns:
s – Updated node embeddings.
- Return type: