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