fix LoDTensor::lod_element to get last element in level

8 years ago · 4018754d66
parent 6a03a4d910
commit 4018754d66
1 changed files with 1 additions and 1 deletions
--- 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];
  }