def testSliceHiddenOnes(self): hidden_size = 60 block_dim = 20 num_blocks = 3 x = tf.ones(shape=[1, 1, hidden_size], dtype=tf.float32) x_sliced = discretization.slice_hidden(x, hidden_size, num_blocks) x_sliced_eval = self.evaluate(x_sliced) self.assertEqual(np.shape(x_sliced_eval), (1, 1, num_blocks, block_dim)) self.assertTrue(np.all(x_sliced_eval == 1))
def testSliceHiddenZeros(self): model_d = 60 block_dim = 20 num_blocks = 3 x = tf.zeros(shape=[1, 1, model_d], dtype=tf.float32) x_sliced = discretization.slice_hidden(x, model_d, num_blocks) x_sliced_eval = self.evaluate(x_sliced) self.assertEqual(np.shape(x_sliced_eval), (1, 1, num_blocks, block_dim)) self.assertTrue(np.all(x_sliced_eval == 0))
def testSliceHiddenOnes(self): hidden_size = 60 block_dim = 20 num_blocks = 3 x = tf.ones(shape=[1, hidden_size], dtype=tf.float32) x_sliced = discretization.slice_hidden(x, hidden_size, num_blocks) with self.test_session() as sess: tf.global_variables_initializer().run() x_sliced_eval = sess.run(x_sliced) self.assertEqual(np.shape(x_sliced_eval), (num_blocks, block_dim)) self.assertTrue(np.all(x_sliced_eval == 1))
def testSliceHiddenOnes(self): hidden_size = 60 block_dim = 20 num_blocks = 3 x = tf.ones(shape=[1, hidden_size], dtype=tf.float32) x_sliced = discretization.slice_hidden(x, hidden_size, num_blocks) with self.test_session() as sess: tf.global_variables_initializer().run() x_sliced_eval = sess.run(x_sliced) self.assertEqual(np.shape(x_sliced_eval), (1, num_blocks, block_dim)) self.assertTrue(np.all(x_sliced_eval == 1))