code: def compute_samples(self):
          
          shape = [self.batch_size, self.eq_samples, self.iw_samples, self.num_latent]
  
          z = tf.nn.softmax(self.pi)
          return z
docs: Sample from a Normal distribution with inferred mu and std