def METHOD_NAME(
    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.METHOD_NAME(
        ids_to_fetch, time, reinit, weight_init_max, weight_init_min
    )