Belle II Software development
GATGAPModel Member List

This is the complete list of members for GATGAPModel, including all inherited members.

__init__(self, units=128, num_features=8, num_pdg=NUM_PDG, emb_size=8, attention_heads=4, n_layers=5, use_gap=False)GATGAPModel
fc_outputGATGAPModel
forward(self, graph)GATGAPModel
gat_layersGATGAPModel
pdg_embeddingGATGAPModel