Method get_local_indices

python/dgl/partition.py:569–573

Get the set of global indices in this given partition.

(self, part, ctx)

        return self._partition

    def get_local_indices(self, part, ctx):
        """Get the set of global indices in this given partition."""
        return self.map_to_global(
            F.arange(0, self.local_size(part), ctx=ctx), part
        )

    def local_size(self, part):
        """Get the number of rows/items assigned to the given part."""