jax
f36d858c - Require shape = sample_shape + batch_shape in random.categorical

Commit

6 years ago

Require shape = sample_shape + batch_shape in random.categorical

References

#1855 - Add categorical sampler

Author

juliuskunze

juliuskunze

Parents

Loading