![]() |
Belle II Software release-09-00-03
|
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_prob | NodeLayer | |
| forward(self, x, edge_index, edge_attr, u, batch) | NodeLayer | |
| lin_in | NodeLayer | |
| lin_out | NodeLayer | |
| lins_hid | NodeLayer | |
| nonlin_function | NodeLayer | |
| norm | NodeLayer | |
| normalize | NodeLayer | |
| num_hid_layers | NodeLayer |