Belle II Software development
|
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_output | GATGAPModel | |
forward(self, graph) | GATGAPModel | |
gat_layers | GATGAPModel | |
pdg_embedding | GATGAPModel |