Как я могу инициализировать переменные с распределением Бернулли в TensorFlow?

Как я могу инициализировать переменные в TensorFlow?

Я хочу связать каждый вес с распределением Бернулли:

• с вероятностью p получить некоторое значение x1, и
• с вероятностью 1-p получить значение x2.

Как мне инициализировать эту матрицу?

Я написал этот код:

logits_y = tf.get_variable("logits", [n_input*n_hidden,2],
                           initializer=tf.constant_initializer(1.))

2 в [n_input*n_hidden, 2] означает [p, 1-p].