create_overlapping_patches

create_overlapping_patches(graph, partition_tensor: LongTensor, patch_graph, min_overlap, target_overlap)

Create overlapping patches from a hard partition of an input graph

Parameters:

• graph – input graph

• partition_tensor – partition of input graph

• patch_graph – graph where nodes are clusters of partition and an edge indicates that the corresponding patches in the output should have at least min_overlap nodes in common

• min_overlap – minimum overlap for connected patches

• target_overlap – maximum overlap during expansion for an edge (additional overlap may result from expansion of other edges)

Returns:

list of node-index tensors for patches