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:

torch.Tensor