fix LoDTensor::lod_element to get last element in level

wanghaoshuang · wanghaoshuang · commit 4018754d666b · 2017-10-19T10:03:47.000+08:00
diff --git 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];
   }
 

Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,7 @@ class LoDTensor : public Tensor {`
`78`	`78`	`*/`
`79`	`79`	`size_t lod_element(size_t level, size_t elem) const {`
`80`	`80`	`PADDLE_ENFORCE_LT(level, NumLevels());`
`81`		`- PADDLE_ENFORCE_LT(elem, NumElements(level));`
	`81`	`+ PADDLE_ENFORCE_LE(elem, NumElements(level));`
`82`	`82`	`return (lod_)[level][elem];`
`83`	`83`	`}`
`84`	`84`