Использование нормального распределения вместо распределения Бернулли

Question

Я новичок в Tensorflow и машинного обучения. Мне нужно использовать нормальное распределение с вариационным автоэнкодером. Я искал примеры, которые используются Нормальное распределение без удачи. Кто-нибудь может привести пример нормального распределения выходных значений в нейронной сети?

  def make_decoder(code, data_shape):

  x = code
  x = tf.layers.dense(x, 200, tf.nn.relu)
  x = tf.layers.dense(x, 200, tf.nn.relu)
 logit = tf.layers.dense(x, np.prod(data_shape))
 logit = tf.reshape(logit, [-1] + data_shape)
 return tfd.Independent(tfd.Bernoulli(logit), 2)