def testImagesToSequenceDims(self):
with self.test_session():
inputs = tf.constant(_rand(2, 7, 11, 5))
outputs = lstm2d.images_to_sequence(inputs)
tf.initialize_all_variables().run()
result = outputs.eval()
self.assertEqual(tuple(result.shape), (11, 14, 5))
def testImagesAndSequenceDims(self):
with self.test_session():
size = (2, 7, 11, 5)
inputs = tf.constant(_rand(*size))
sequence = lstm2d.images_to_sequence(inputs)
outputs = lstm2d.sequence_to_images(sequence, size[0])
tf.initialize_all_variables().run()
result = outputs.eval()
self.assertEqual(tuple(result.shape), size)
