Belle II Software light-2406-ragdoll
NodeLayer Member List

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

__init__(self, nfeat_in_dim, efeat_in_dim, gfeat_in_dim, nfeat_hid_dim, nfeat_out_dim, num_hid_layers, dropout, normalize=True)NodeLayer
dropout_probNodeLayer
forward(self, x, edge_index, edge_attr, u, batch)NodeLayer
lin_inNodeLayer
lin_outNodeLayer
lins_hidNodeLayer
nonlin_functionNodeLayer
normNodeLayer
normalizeNodeLayer
num_hid_layersNodeLayer