Fix unitttest in sequence_pad_op

upload-readme
Yibing Liu 7 years ago
parent 9e3245e3bd
commit ce773ed7f8

@ -62,7 +62,7 @@ class TestSequencePadOp(OpTest):
start_idx = end_idx start_idx = end_idx
out_data = np.array(padded_sequences) out_data = np.array(padded_sequences)
length = np.array(self.x_len_lod[0]) length = np.array(self.x_len_lod[0]).reshape((-1, 1))
self.outputs = {'Out': out_data, 'Length': length} self.outputs = {'Out': out_data, 'Length': length}
def setUp(self): def setUp(self):
@ -130,3 +130,7 @@ class TestSequencePadOp7(TestSequencePadOp):
self.pad_value = [1.0] self.pad_value = [1.0]
self.padded_length = 7 self.padded_length = 7
self.dtype = 'float32' self.dtype = 'float32'
if __name__ == '__main__':
unittest.main()

Loading…
Cancel
Save