tfa.seq2seq.sampler.categorical_sample

View source on GitHub

Samples from categorical distribution.