Sample from a Normal distribution with inferred mu and std

def compute_samples(self):
        """ Sample from a Normal distribution with inferred mu and std """
        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