Как сгенерировать тензор Бернулли в тензорном потоке

Question

Как я могу сгенерировать тензор в tensorflow из Bernoulli distribution с коэффициентом p?

Например:

a = tf.bernoulli(shape=[10,10], p)

создает матрицу 10x10 из 0-1, где каждый элемент матрицы равен единице с вероятностью p и нулю с вероятностью 1-p.

Answer 1

Я могу решить мою проблему! : D Следующий код генерирует, что мне нужно где p=0.7:

p = tf.constant([0.7])
r = tf.random.uniform(shape=shape, maxval=1)
b = tf.math.greater(p, r)
f = tf.cast(b, dtype=tf.float32)