Fix the index check in ndarray. (apache#4398)

N-length array should only have index from 0 to N-1.
iblislin · Dec 29, 2016 · 6f95b74 · 6f95b74
1 parent 835d690
commit 6f95b74
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/mxnet/ndarray.h b/include/mxnet/ndarray.h
@@ -256,7 +256,7 @@ class NDArray {
   inline NDArray At(index_t idx) const {
     NDArray ret = *this;
     CHECK(!is_none()) << "NDArray is not initialized";
-    CHECK_GE(shape_[0], idx) << "index out of range";
+    CHECK_GT(shape_[0], idx) << "index out of range";
     size_t length = shape_.ProdShape(1, shape_.ndim());
     ret.offset_ += idx * length;
     if (shape_.ndim() > 1) {