code: def __call__(self, logits: torch.Tensor):
      """
      Sample the most likely token from the distribution of logits
      """
      return logits.argmax(dim=-1)
  
imports: import torch
  from labml_nn.sampling import Sampler
label_desc: Use image node_id 1 for calling the GreedySampler obj's underlying member method code with example usage: obj.__call__(logits) and returns: logits