|
|
@ -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()
|
|
|
|