code: def __init__(self, size):
      self._max_size = size
      self._current_size = 0
      self._bucket = []
  
file_path: ColossalAI/colossalai/zero/low_level/bookkeeping/tensor_bucket.py
imports: from torch._utils import _flatten_dense_tensors, _unflatten_dense_tensors
label_desc: Use image node_id 1 to create a new TensorBucket object  with example: obj = TensorBucket(size)