From e4994fba19644c7e09cd10a614118d2fcb343b6f Mon Sep 17 00:00:00 2001
From: Malinda <malinda.dilhara@gmail.com>
Date: Fri, 1 Jul 2022 01:54:45 -0700
Subject: [PATCH] use np.cumsum

---
 ADE/ade/reader.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/ADE/ade/reader.py b/ADE/ade/reader.py
index d3e2f95..632dd6d 100755
--- a/ADE/ade/reader.py
+++ b/ADE/ade/reader.py
@@ -104,9 +104,7 @@ def create_lodtensor(data_ids, place):
             cur_len = 0
             lod = [cur_len]
             seq_lens = [len(ids) for ids in data_ids]
-            for l in seq_lens: 
-                cur_len += l
-                lod.append(cur_len)
+            lod = np.sumsum(seq_lens)
             flattened_data = np.concatenate(data_ids, axis=0).astype("int64")
             flattened_data = flattened_data.reshape([len(flattened_data), 1])
             res = fluid.LoDTensor()