graphinvent/gnn/summation_mpnn.py
def readout(self, hidden_nodes : torch.Tensor, input_nodes : torch.Tensor,
                node_mask : torch.Tensor) -> None:
        """
        Local readout function, to be implemented in all `SummationMPNN` subclasses.

        Args:
        ----
            hidden_nodes (torch.Tensor) : Batch of node feature vectors.
            input_nodes (torch.Tensor) : Batch of node feature vectors.
            node_mask (torch.Tensor) : Mask for non-existing neighbors, where elements
                                       are 1 if corresponding element exists and 0
                                       otherwise.

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

        Local readout function, to be implemented in all `SummationMPNN` subclasses.

        Args:
        ----
            hidden_nodes (torch.Tensor) : Batch of node feature vectors.
            input_nodes (torch.Tensor) : Batch of node feature vectors.
            node_mask (torch.Tensor) : Mask for non-existing neighbors, where elements
                                       are 1 if corresponding element exists and 0
                                       otherwise.

        Shapes:
        ------
            hidden_nodes : (total N nodes in batch, N node features)
            input_nodes : (total N nodes in batch, N node features)
            node_mask : (total N nodes in batch, N features)
        
Local readout function, to be implemented in all `SummationMPNN` subclasses.
Local readout function to be implemented in all ` SummationMPNN ` subclasses .
def readout(self, hidden_nodes : torch.Tensor, input_nodes : torch.Tensor,
                node_mask : torch.Tensor) -> None:
        raise NotImplementedError
def readout ( self , hidden_nodes : torch . Tensor , input_nodes : torch . Tensor , node_mask : torch . Tensor ) -> None : raise NotImplementedError
Local readout function, to be implemented in all `SummationMPNN` subclasses.
Local readout function to be implemented in all ` SummationMPNN ` subclasses .
"""
        Local readout function, to be implemented in all `SummationMPNN` subclasses.

        Args:
        ----
            hidden_nodes (torch.Tensor) : Batch of node feature vectors.
            input_nodes (torch.Tensor) : Batch of node feature vectors.
            node_mask (torch.Tensor) : Mask for non-existing neighbors, where elements
                                       are 1 if corresponding element exists and 0
                                       otherwise.

        Shapes:
        ------
            hidden_nodes : (total N nodes in batch, N node features)
            input_nodes : (total N nodes in batch, N node features)
            node_mask : (total N nodes in batch, N features)
        """
self
hidden_nodes
torch.Tensor
Batch of node feature vectors.
Batch of node feature vectors .
input_nodes
torch.Tensor
Batch of node feature vectors.
Batch of node feature vectors .
node_mask
torch.Tensor
Mask for non-existing neighbors, where elements
are 1 if corresponding element exists and 0
otherwise.
Mask for non - existing neighbors where elements are 1 if corresponding element exists and 0 otherwise .