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

tfa.seq2seq.sampler.bernoulli_sample

View source on GitHub

Samples from Bernoulli distribution.

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