def testProjectHidden(self): hidden_size = 60 block_dim = 20 num_blocks = 3 x = tf.zeros(shape=[1, 1, hidden_size], dtype=tf.float32) projection_tensors = tf.random_normal( shape=[num_blocks, hidden_size, block_dim], dtype=tf.float32) x_projected = discretization.project_hidden(x, projection_tensors, hidden_size, num_blocks) x_projected_eval = self.evaluate(x_projected) self.assertEqual(np.shape(x_projected_eval), (1, 1, num_blocks, block_dim)) self.assertTrue(np.all(x_projected_eval == 0))
def testProjectHidden(self): hidden_size = 60 block_dim = 20 num_blocks = 3 x = tf.zeros(shape=[1, hidden_size], dtype=tf.float32) projection_tensors = tf.random_normal( shape=[num_blocks, hidden_size, block_dim], dtype=tf.float32) x_projected = discretization.project_hidden(x, projection_tensors, hidden_size, num_blocks) with self.test_session() as sess: tf.global_variables_initializer().run() x_projected_eval = sess.run(x_projected) self.assertEqual(np.shape(x_projected_eval), (1, num_blocks, block_dim)) self.assertTrue(np.all(x_projected_eval == 0))