tensorflow - 使用正态分布而不是伯努利分布

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)