提交 ce773ed7 编写于 作者: Y Yibing Liu

Fix unitttest in sequence_pad_op

上级 9e3245e3
...@@ -62,7 +62,7 @@ class TestSequencePadOp(OpTest): ...@@ -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): ...@@ -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()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册