code: def compute_samples(self):
          
          eps = tf.random_normal([self.batch_size, self.eq_samples, self.iw_samples, self.num_latent])
          z = tf.reshape(eps * self.std + self.mu, [-1, self.num_latent])
          return z
docs: Sample from a Normal distribution with inferred mu and std