Announcing the TensorFlow Dev Summit 2020 Learn more

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
)