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, 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))
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))
