def sample_gaussian(x, sigma): return x + tf.random_normal(tf.shape(x), mean=0.0, stddev=sigma, dtype=tf.float32)