Marcello Fuschi b8d460d203 Add Tensor.multinomial (#2295) ...

* add Tensor.multinomial only with replacement

* add support for 2D input in Tensor.multinomial

* fix multinomial output shape

* allow passing replacement=False to Tensor.multinomial when num_samples=1

* improve tests for Tensor.multinomial

* fix edge case in Tensor.multinomial

* Tensor.multinomial no more staticmethod

2023-11-15 11:38:39 -08:00

6.6 KiB

Raw Blame History

View Raw