torch.distributions.Categorical в C++

Question

Есть ли C++ API с такой же функциональностью, как torch.distributions.Categorical? я нашел torch::::multinomial с одинаковыми входными параметрами (const Tensor &self, int64_t num_samples, bool replacement = false, Generator *generator = nullptr), но это не имеет sample или же log_prob API-интерфейсы.

Спасибо афшин