probly.representer.sngp.torch.torch_compute_categorical_sample_from_logits¶

probly.representer.sngp.torch.torch_compute_categorical_sample_from_logits(sample: TorchSample[Any]) → TorchCategoricalDistributionSample[source]¶: Convert a TorchSample of SNGP logits to a categorical distribution sample.