diff --git a/paddle/framework/lod_tensor.h b/paddle/framework/lod_tensor.h index 4db36ee76609ac6360fe2fc7b4a366e0284d1016..3eab91b0d1c6b8fd471e851872d2c3263a7df59e 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]; }