def test_make_z_normal(self): """Tests the function that makes the latent variable tensors.""" if tf.executing_eagerly(): return zs = generator.make_z_normal(2, 16, 128) sess = tf.train.MonitoredTrainingSession() z_batch = sess.run(zs) self.assertEqual((2, 16, 128), z_batch.shape)
def _make_noise(_): noise = gen_module.make_z_normal(num_towers, bs, params['z_dim']) return noise[0] # one tower