fix icafe issue: DLTP-3138 (sequence_unpad compatibility) (#3892)

5fee7182 · Meiyim · pkpk · 207c1969 · 5fee7182
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

PaddleNLP/preprocess/padding.py PaddleNLP/preprocess/padding.py +1 -1

未找到文件。
--- a/PaddleNLP/preprocess/padding.py
+++ b/PaddleNLP/preprocess/padding.py
@@ -69,7 +69,7 @@ def pad_batch_data(insts,
    if return_seq_lens:
        seq_lens = np.array([len(inst) for inst in insts])
-        return_list += [seq_lens.astype("int64").reshape([-1, 1])]
+        return_list += [seq_lens.astype("int64").reshape([-1])]
    return return_list if len(return_list) > 1 else return_list[0]