probly.representer.sngp.torch

Torch specific functionality for the SNGP representer.

Functions

torch_compute_categorical_sample_from_logits

Convert a TorchSample of SNGP logits to a categorical distribution sample.