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