diff --git a/src/operator/numpy/np_matrix_op-inl.h b/src/operator/numpy/np_matrix_op-inl.h index b81f2ffee481..2007a49ca662 100644 --- 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 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) {