docstr: Record the padding size of a param
  
  Args:
      param (Tensor): The parameter
      padding_size (int): The padding size of the parameter
docstr_tok: ["Record","the","padding","size","of","a","param","Args",":","param","(","Tensor",")",":","The","parameter","padding_size","(","int",")",":","The","padding","size","of","the","parameter"]
code: def record_param_padding_size(self, param: Tensor, padding_size: int):
      """Record the padding size of a param
  
      Args:
          param (Tensor): The parameter
          padding_size (int): The padding size of the parameter
      """
  
      self._padding_map[id(param)] = padding_size
  
file_path: colossalai/colossalai/zero/low_level/bookkeeping/parameter_store.py
imports: from torch import Tensor
  from torch.distributed import ProcessGroup
  from .base_store import BaseStore
label_desc: Use image node_id 2 for calling the ParameterStore obj's underlying member method code with example usage: obj.record_param_padding_size(param, padding_size) without return types