tf.random.poisson

View source on GitHub

Draws shape samples from each of the given Poisson distribution(s).

tf.random.poisson(
    shape, lam, dtype=tf.dtypes.float32, seed=None, name=None
)

lam is the rate parameter describing the distribution(s).

Example:

samples = tf.random.poisson([10], [0.5, 1.5])
# samples has shape [10, 2], where each slice [:, 0] and [:, 1] represents
# the samples drawn from each distribution

samples = tf.random.poisson([7, 5], [12.2, 3.3])
# samples has shape [7, 5, 2], where each slice [:, :, 0] and [:, :, 1]
# represents the 7x5 samples drawn from each of the two distributions

Args:

Returns: