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
)