torch.distributions.Categorical в C ++ - PullRequest
0 голосов
/ 20 мая 2019

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

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

...