class_docstr: _FakePalettizationTensorHook is the class to assist in using CPU when we only want to utilize a certain percentage
  of the GPU memory.
code: def __init__(
      self,
      size_list,
      use_cpu: bool = False,
      name: str = None,
      palett_tau: float = 0.0001,
  ):
      self.name = name
      self.size_list = size_list
      self.tensor_list = [None] * len(size_list)
      self.device_list = [None] * len(size_list)
      self.use_cpu = use_cpu
      self.palett_tau = palett_tau
  
file_path: coremltools/coremltools/optimize/torch/palettization/_fake_palettizer_tensor_hook.py
imports: import torch
  import torch.nn.functional
label_desc: Use image node_id 1 to create a new _FakePalettizationTensorHook object  with example: obj = _FakePalettizationTensorHook(size_list, use_cpu, name, palett_tau)