gregate on. Returns: A Tuple of the followings: A List[`Shard`] which contains the local tensor and its metadata. A List[`ShardMetadata`] which contains the metadata for the shard, including offsets, lengths and device placement. r