TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tfa.seq2seq.sampler.categorical_sample

View source on GitHub

Samples from categorical distribution.

tfa.seq2seq.sampler.categorical_sample(
    logits,
    dtype=tf.int32,
    sample_shape=(),
    seed=None
)