කෑදරනියැදීම

14import torch
15
16from labml_nn.sampling import Sampler

19class GreedySampler(Sampler):

20    def __call__(self, logits: torch.Tensor):

24        return logits.argmax(dim=-1)