def sample_likelihood(self, zs): """x | p ~ p(x | p)""" return {'x': bernoulli.sample(p=tf.ones(10) * zs['p'])}