class_docstr: A manager class to manipulate the tensors in chunks.
  
  Args:
      chunk_configuration (Dict[int, Dict]): the configuration dictionary of this chunk manager.
      init_device (torch.device): optional, the device on which the chunk is initialized. The default is None.
class_docstr_tok: ["A","manager","class","to","manipulate","the","tensors","in","chunks",".","Args",":","chunk_configuration","(","Dict","[","int",",","Dict","]",")",":","the","configuration","dictionary","of","this","chunk","manager",".","init_device","(","torch.device",")",":","optional",",","the","device","on","which","the","chunk","is","initialized",".","The","default","is","None","."]
docstr: Return the chunk owning the tensor.
  
  Args:
      tensor (torch.Tensor): a torch tensor object
docstr_tok: ["Return","the","chunk","owning","the","tensor",".","Args",":","tensor","(","torch.Tensor",")",":","a","torch","tensor","object"]
code: def get_chunk(self, tensor: torch.Tensor) -> Chunk:
      """
      Return the chunk owning the tensor.
  
      Args:
          tensor (torch.Tensor): a torch tensor object
      """
      return self.tensor_chunk_map[tensor]
  
imports: from collections import deque
  from typing import Deque, Dict, Iterable, List, Optional, Set, Tuple
  import torch
  import torch.distributed
  from torch.distributed import ProcessGroup
  from colossalai.utils import free_storage, get_current_device
  from .chunk import Chunk, ChunkFullError, TensorState
label_desc: Use image node_id 11 for calling the ChunkManager obj's underlying member method code with example usage: obj.get_chunk(tensor) and returns: self