def METHOD_NAME(self, ids):
    if isinstance(ids, int):
        return self.decoder[ids]
    if isinstance(ids, torch.Tensor):
        ids = ids.cpu().tolist()
    return [self.decoder[idx] for idx in ids]