U yhP@s^ddlZddlmZddlmZdZejeeeeejdddZejeejdd d Z dS) N) ShardMetadata)Sequencez@Please use DTensor instead and we are deprecating ShardedTensor.)tensoroffsetssizesreturncCs@|}tt||D](\}\}}|||kr||||}q|S)zC Narrow the tensor according to ``offsets`` and ``sizes``. ) enumeratezipsizeZnarrow)rrrZnarrowed_tensoridxoffsetr r Q/var/www/html/venv/lib/python3.8/site-packages/torch/distributed/_shard/_utils.pynarrow_tensor_by_indexsr)rmetadatarcCst||j|jS)z5 Narrow the tensor according to the metadata )rZ shard_offsetsZ shard_sizes)rrr r r narrow_tensorsr) ZtorchZ!torch.distributed._shard.metadatartypingrZ DEPRECATE_MSGZTensorintrrr r r rs