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

        Args:
        ----
            nodes (torch.Tensor)           : Batch of node feature vectors.
            node_neighbours (torch.Tensor) : Batch of node feature vectors for neighbors.
            edges (torch.Tensor)           : Batch of edge feature vectors.

        Shapes:
        ------
            nodes           : (total N nodes in batch, N node features)
            node_neighbours : (total N nodes in batch, max node degree, N node features)
            edges           : (total N nodes in batch, max node degree, N edge features)
        """
        raise NotImplementedError

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

        Args:
        ----
            nodes (torch.Tensor)           : Batch of node feature vectors.
            node_neighbours (torch.Tensor) : Batch of node feature vectors for neighbors.
            edges (torch.Tensor)           : Batch of edge feature vectors.

        Shapes:
        ------
            nodes           : (total N nodes in batch, N node features)
            node_neighbours : (total N nodes in batch, max node degree, N node features)
            edges           : (total N nodes in batch, max node degree, N edge features)
        
Message passing function, to be implemented in all `SummationMPNN` subclasses.
Message passing function to be implemented in all ` SummationMPNN ` subclasses .
def message_terms(self, nodes : torch.Tensor, node_neighbours : torch.Tensor,
                      edges : torch.Tensor) -> None:
        raise NotImplementedError
def message_terms ( self , nodes : torch . Tensor , node_neighbours : torch . Tensor , edges : torch . Tensor ) -> None : raise NotImplementedError
Message passing function, to be implemented in all `SummationMPNN` subclasses.
Message passing function to be implemented in all ` SummationMPNN ` subclasses .
"""
        Message passing function, to be implemented in all `SummationMPNN` subclasses.

        Args:
        ----
            nodes (torch.Tensor)           : Batch of node feature vectors.
            node_neighbours (torch.Tensor) : Batch of node feature vectors for neighbors.
            edges (torch.Tensor)           : Batch of edge feature vectors.

        Shapes:
        ------
            nodes           : (total N nodes in batch, N node features)
            node_neighbours : (total N nodes in batch, max node degree, N node features)
            edges           : (total N nodes in batch, max node degree, N edge features)
        """
self
nodes
torch.Tensor
Batch of node feature vectors.
Batch of node feature vectors .
node_neighbours
torch.Tensor
Batch of node feature vectors for neighbors.
Batch of node feature vectors for neighbors .
edges
torch.Tensor
Batch of edge feature vectors.
Batch of edge feature vectors .