Accept result tensor in cat.rnd() with options.categories = true

google-deepmind / torch-distributions

Probability distributions, wrapped for Torch.

BSD 3-Clause "New" or "Revised" License

61 stars 25 forks source link

Accept result tensor in cat.rnd() with options.categories = true #24

Open jucor opened 10 years ago

jucor commented 10 years ago

At the moment, cat.rnd() does not use the result tensor if categories == true, because torch.index() doesn't -- see torch/torch7-distro#202. All other cases are fine.

jucor commented 9 years ago

Note: torch.index() now does, nothing stands in the way, except time to do it.