tensorflow - 如何在tensorflow中生成伯努利张量

Question

如何生成带有因子 的张tensorflow量？Bernoulli distributionp

例如：

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

生成一个10x100-1 的矩阵，其中矩阵的每个元素为 1 概率p和 0 概率1-p.

标签： tensorflowbernoulli-probability

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)