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