graphinvent/gnn/summation_mpnn.py
def update(self, nodes : torch.Tensor, messages : torch.Tensor) -> None:
        """
        Message update function, to be implemented in all `SummationMPNN` subclasses.

        Args:
        ----
            nodes (torch.Tensor)    : Batch of node feature vectors.
            messages (torch.Tensor) : Batch of incoming messages.

        Shapes:
        ------
            nodes    : (total N nodes in batch, N node features)
            messages : (total N nodes in batch, N node features)
        """
        raise NotImplementedError

        Message update function, to be implemented in all `SummationMPNN` subclasses.

        Args:
        ----
            nodes (torch.Tensor)    : Batch of node feature vectors.
            messages (torch.Tensor) : Batch of incoming messages.

        Shapes:
        ------
            nodes    : (total N nodes in batch, N node features)
            messages : (total N nodes in batch, N node features)
        
Message update function, to be implemented in all `SummationMPNN` subclasses.
Message update function to be implemented in all ` SummationMPNN ` subclasses .
def update(self, nodes : torch.Tensor, messages : torch.Tensor) -> None:
        raise NotImplementedError
def update ( self , nodes : torch . Tensor , messages : torch . Tensor ) -> None : raise NotImplementedError
Message update function, to be implemented in all `SummationMPNN` subclasses.
Message update function to be implemented in all ` SummationMPNN ` subclasses .
"""
        Message update function, to be implemented in all `SummationMPNN` subclasses.

        Args:
        ----
            nodes (torch.Tensor)    : Batch of node feature vectors.
            messages (torch.Tensor) : Batch of incoming messages.

        Shapes:
        ------
            nodes    : (total N nodes in batch, N node features)
            messages : (total N nodes in batch, N node features)
        """
self
nodes
torch.Tensor
Batch of node feature vectors.
Batch of node feature vectors .
messages
torch.Tensor
Batch of incoming messages.
Batch of incoming messages .