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)