code: def __init__(self, tensor, group, partition_meta=None):
      super().__init__()
  
      self.group = group
      self.num_parts = dist.get_world_size(group=self.group)
      self.rank = dist.get_rank(group=self.group)
  
      self.orig_size = list(tensor.size())
      self.orig_device = tensor.device
      self.local_data, self.partition = self._partition_tensor(tensor)
  
imports: from collections.abc import Iterable
  from deepspeed.moe.utils import is_moe_param
  import os
  import psutil
  import gc
  from math import sqrt
  from math import floor
  from bisect import bisect_left
  import torch
  from torch import inf
  from deepspeed import comm
  from deepspeed.utils import groups, logger
  from deepspeed.runtime.constants import PIPE_REPLICATED
  from numpy import prod
label_desc: Use image node_id 1 to create a new PartitionedTensor object  with example: obj = PartitionedTensor(tensor, group, partition_meta)