enerzyme.models.leftnet.core.aggregate_pos#

class enerzyme.models.leftnet.core.aggregate_pos(aggr='mean')[source]#

Bases: MessagePassing

__init__(aggr='mean')[source]#: Initialize internal Module state, shared by both nn.Module and ScriptModule.

forward(vector, edge_index)[source]#: Runs the forward pass of the module.