sg%ddlmZddlZddlmZdZdej deedeedej fd Zdej d edej fd Z y) )SequenceN) ShardMetadataz@Please use DTensor instead and we are deprecating ShardedTensor.tensoroffsetssizesreturnc|}tt||D]0\}\}}||j|ks|j|||}2|S)zC Narrow the tensor according to ``offsets`` and ``sizes``. ) enumeratezipsizenarrow)rrrnarrowed_tensoridxoffsetr s R/var/www/html/venv/lib/python3.12/site-packages/torch/distributed/_shard/_utils.pynarrow_tensor_by_indexr s^O(We)<=H^fd &++c" ".44S&$GO H metadatacDt||j|jS)z5 Narrow the tensor according to the metadata )r shard_offsets shard_sizes)rrs r narrow_tensorrs "&(*@*@(BVBV WWr) typingrtorch!torch.distributed._shard.metadatar DEPRECATE_MSGTensorintrrrrr su ;S  LL c] C= \\ $X%,,X-XELLXr