Skip to content

sampler

[ allennlp.nn.samplers.sampler ]


Sampler#

class Sampler(Registrable)

An abstract class representing a multinomial sampler

__call__#

class Sampler(Registrable):
 | ...
 | def __call__(
 |     self,
 |     logits: torch.Tensor,
 |     num_samples: int = 1,
 |     with_replacement: bool = True
 | ) -> torch.Tensor