docstr: Fetch `ids_to_fetch` from tensor. If `reinit` is set to `True`, will
  reinitialize the embedding if the global id is not in PS.
docstr_tok: ["Fetch","`","ids_to_fetch","`","from","tensor",".","If","`","reinit","`","is","set","to","`","True","`",",","will","reinitialize","the","embedding","if","the","global","id","is","not","in","PS","."]
code: def fetch(
      self,
      ids_to_fetch: torch.Tensor,
      time: int,
      reinit: bool = False,
      weight_init_max: float = 0,
      weight_init_min: float = 0,
  ):
      """
      Fetch `ids_to_fetch` from tensor. If `reinit` is set to `True`, will
      reinitialize the embedding if the global id is not in PS.
      """
      return self._ps.fetch(
          ids_to_fetch, time, reinit, weight_init_max, weight_init_min
      )
  
file_path: torchrec/contrib/dynamic_embedding/src/torchrec_dynamic_embedding/ps.py
imports: import os
  from typing import Callable, Dict, List, Tuple, Union
  import torch
  from torch.distributed._shard.sharded_tensor import ShardedTensor
  from torchrec.distributed.types import ParameterSharding
  from .tensor_list import TensorList
label_desc: Use image node_id 3 for calling the PS obj's underlying member method code with example usage: obj.fetch(ids_to_fetch, time, reinit, weight_init_max, weight_init_min) and returns: self