Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

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
)