graphinvent/gnn/edge_mpnn.py
def propagate_edges(self, edges : torch.Tensor, ingoing_edge_memories : torch.Tensor,
                        ingoing_edges_mask : torch.Tensor) -> None:
        """
        Edge propagation rule, to be implemented in all `EdgeMPNN` subclasses.

        Args:
        ----
            edges (torch.Tensor)                 : Batch of edge feature tensors.
            ingoing_edge_memories (torch.Tensor) : Batch of memories for all
                                                   ingoing edges.
            ingoing_edges_mask (torch.Tensor)    : Mask for ingoing edges.

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

        Edge propagation rule, to be implemented in all `EdgeMPNN` subclasses.

        Args:
        ----
            edges (torch.Tensor)                 : Batch of edge feature tensors.
            ingoing_edge_memories (torch.Tensor) : Batch of memories for all
                                                   ingoing edges.
            ingoing_edges_mask (torch.Tensor)    : Mask for ingoing edges.

        Shapes:
        ------
            edges                 : (batch size, N nodes, N nodes, total N edge features)
            ingoing_edge_memories : (total N edges in batch, total N edge features)
            ingoing_edges_mask    : (total N edges in batch, max node degree, total N edge features)
        
Edge propagation rule, to be implemented in all `EdgeMPNN` subclasses.
Edge propagation rule to be implemented in all ` EdgeMPNN ` subclasses .
def propagate_edges(self, edges : torch.Tensor, ingoing_edge_memories : torch.Tensor,
                        ingoing_edges_mask : torch.Tensor) -> None:
        raise NotImplementedError
def propagate_edges ( self , edges : torch . Tensor , ingoing_edge_memories : torch . Tensor , ingoing_edges_mask : torch . Tensor ) -> None : raise NotImplementedError
Edge propagation rule, to be implemented in all `EdgeMPNN` subclasses.
Edge propagation rule to be implemented in all ` EdgeMPNN ` subclasses .
"""
        Edge propagation rule, to be implemented in all `EdgeMPNN` subclasses.

        Args:
        ----
            edges (torch.Tensor)                 : Batch of edge feature tensors.
            ingoing_edge_memories (torch.Tensor) : Batch of memories for all
                                                   ingoing edges.
            ingoing_edges_mask (torch.Tensor)    : Mask for ingoing edges.

        Shapes:
        ------
            edges                 : (batch size, N nodes, N nodes, total N edge features)
            ingoing_edge_memories : (total N edges in batch, total N edge features)
            ingoing_edges_mask    : (total N edges in batch, max node degree, total N edge features)
        """
self
edges
torch.Tensor
Batch of edge feature tensors.
Batch of edge feature tensors .
ingoing_edge_memories
torch.Tensor
Batch of memories for all
ingoing edges.
Batch of memories for all ingoing edges .
ingoing_edges_mask
torch.Tensor
Mask for ingoing edges.
Mask for ingoing edges .