sar.common_tuples.ShardEdgesAndFeatures

class ShardEdgesAndFeatures(edges: Tuple[Tensor, Tensor], edge_features: Dict[str, Tensor])

Stores the edge information for all edges connecting nodes in one partition to nodes in another partition. For an N-way partition, each worker will have N ShardEdgesAndFeatures object, where each object contains data for incoming edges from each partition (including the worker’s own partition)

edges : Tuple[Tensor,Tensor]

The source and destination global node ids for each edge in the shard

edge_features : Dict[str,Tensor]

A dictionary of the edge features