code: def __call__(self, logits: torch.Tensor) -> torch.Tensor:
      """
      ### Sample from logits
  
      :param logits: are the logits of the distribution of shape `[..., n_tokens]`
      """
      raise NotImplementedError()
  
label_desc: Use image node_id 1 for calling the Sampler obj's underlying member method code with example usage: obj.__call__(logits) without return types