def testConvEncoder(self):
sequence_length = [17, 21, 20]
inputs = tf.zeros([3, 21, 5])
encoder = encoders.ConvEncoder(3, 3, 10)
outputs, _, encoded_length = encoder(
inputs, sequence_length=tf.constant(sequence_length))
outputs, encoded_length = self.evaluate([outputs, encoded_length])
self.assertAllEqual([3, 21, 10], outputs.shape)
self.assertAllEqual(sequence_length, encoded_length)
def testConvEncoder(self):
sequence_length = [17, 21, 20]
inputs = _build_dummy_sequences(sequence_length)
encoder = encoders.ConvEncoder(3, 10)
outputs, _, encoded_length = encoder.encode(
inputs, sequence_length=tf.constant(sequence_length))
with self.test_session() as sess:
sess.run(tf.global_variables_initializer())
outputs, encoded_length = sess.run([outputs, encoded_length])
self.assertAllEqual([3, 21, 10], outputs.shape)
self.assertAllEqual(sequence_length, encoded_length)