change error info

apache · Dec 31, 2019 · 0eb9f68 · 0eb9f68
1 parent a3ad246
commit 0eb9f68
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/operator/numpy/np_matrix_op-inl.h b/src/operator/numpy/np_matrix_op-inl.h
@@ -498,7 +498,8 @@ void NumpyFlipForward(const nnvm::NodeAttrs& attrs,
   std::vector<index_t>  trailing_(axistemp.ndim());
   index_t flip_index = 0;
   for (int axis : axistemp) {
-    CHECK_LT(axis, ishape.ndim()) << "Input must be >= " << axis + 1 << "-d";
+    CHECK_LT(axis, ishape.ndim()) << "axis " << axis
+      << " is out of bounds for array of dimension " << ishape.ndim() << std::endl;
     stride_[flip_index] = ishape[axis];
     trailing_[flip_index] = 1;
     for (int i2 = axis + 1; i2 < ishape.ndim(); ++i2) {