Hi,
Is there any C++ API with same functionality as torch.distributions.Categorical?
I found torch::::multinomial that has same input parameters (const Tensor &self, int64_t num_samples, bool replacement = false, Generator *generator = nullptr), but it does not have sample or log_prob APIs.
Thanks,
Afshin