from torch_geometric.nn import GCNConv, GATConv, GINConv, global_add_pool

# GCNConv: Graph Convolutional Network layer
# GATConv: Graph Attention Network layer
# GINConv: Graph Isomorphism Network layer
# global_add_pool: Global pooling operation that sums up all node embeddings in a graph

# These are all neural network layers commonly used in graph neural networks (GNNs) for graph classification tasks. The GCNConv and GATConv layers perform message passing between neighboring nodes in the graph, while the GINConv layer uses a learnable function to aggregate node features. The global_add_pool operation aggregates node embeddings to obtain a fixed-size representation of the entire graph.