From 4018754d666bc8045177e295850d941de4b264df Mon Sep 17 00:00:00 2001
From: wanghaoshuang <wanghaoshuang@baidu.com>
Date: Thu, 19 Oct 2017 10:03:47 +0800
Subject: [PATCH] fix LoDTensor::lod_element to get last element in level

---
 paddle/framework/lod_tensor.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/framework/lod_tensor.h b/paddle/framework/lod_tensor.h
index 4db36ee7660..3eab91b0d1c 100644
--- a/paddle/framework/lod_tensor.h
+++ b/paddle/framework/lod_tensor.h
@@ -78,7 +78,7 @@ class LoDTensor : public Tensor {
    */
   size_t lod_element(size_t level, size_t elem) const {
     PADDLE_ENFORCE_LT(level, NumLevels());
-    PADDLE_ENFORCE_LT(elem, NumElements(level));
+    PADDLE_ENFORCE_LE(elem, NumElements(level));
     return (lod_)[level][elem];
   }
 
-- 
GitLab