使用正态分布代替伯努利分布

时间:2018-01-19 18:11:39

标签: tensorflow tensorflow-probability

我是Tensorflow和机器学习的新手。我需要使用带有变量自动编码器的Normal分布。我搜索了使用正态分布但没有运气的示例。 任何人都可以提供神经网络中输出值正态分布的示例吗?

  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)

0 个答案:

没有答案