Sample from a Normal distribution with inferred mu and std

def compute_samples(self):
        """ Sample from a Normal distribution with inferred mu and std """
        shape = [self.batch_size, self.eq_samples, self.iw_samples, self.num_latent]

        z = tf.nn.softmax(self.pi)
        return z