terminator.models.layers.energies.gvp.EdgeLayer

class terminator.models.layers.energies.gvp.EdgeLayer(node_dims, edge_dims, drop_rate=0.1, n_layers=3, module_list=None, activations=(<function relu>, <built-in method sigmoid of type object>), vector_gate=False)

Bases: Module

GVP Edge MPNN

__init__(node_dims, edge_dims, drop_rate=0.1, n_layers=3, module_list=None, activations=(<function relu>, <built-in method sigmoid of type object>), vector_gate=False)

Initializes internal Module state, shared by both nn.Module and ScriptModule.

Methods

__init__(node_dims, edge_dims[, drop_rate, ...])	Initializes internal Module state, shared by both nn.Module and ScriptModule.
forward(h_V, edge_index, h_E[, node_mask])	TODO

Attributes

T_destination	alias of TypeVar('T_destination', bound=Mapping[str, Tensor])
dump_patches	This allows better BC support for load_state_dict().

forward(h_V, edge_index, h_E, node_mask=None)

TODO