Belle II Software light-2406-ragdoll
|
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 |