Belle II Software light-2406-ragdoll
EdgeLayer Member List

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

__init__(self, nfeat_in_dim, efeat_in_dim, gfeat_in_dim, efeat_hid_dim, efeat_out_dim, num_hid_layers, dropout, normalize=True)EdgeLayer
dropout_probEdgeLayer
forward(self, src, dest, edge_attr, u, batch)EdgeLayer
lin_inEdgeLayer
lin_outEdgeLayer
lins_hidEdgeLayer
nonlin_functionEdgeLayer
normEdgeLayer
normalizeEdgeLayer
num_hid_layersEdgeLayer