def tf_normal(y, mu, sigma):
oneDivSqrtTwoPI = 1 / math.sqrt(2*math.pi)
result = tf.sub(y, mu)
result = tf.transpose(result, [2,1,0])
result = tf.mul(result,tf.inv(sigma + 1e-8))
result = -tf.square(result)/2
result = tf.mul(tf.exp(result),tf.inv(sigma + 1e-8))*oneDivSqrtTwoPI
result = tf.reduce_prod(result, reduction_indices=[0])
return result
评论列表
文章目录