diff --git a/src/operator/tensor/init_op.h b/src/operator/tensor/init_op.h index 2cdf046b36b1..93fcd4ee6091 100644 --- a/src/operator/tensor/init_op.h +++ b/src/operator/tensor/init_op.h @@ -134,7 +134,7 @@ struct RangeParam : public dmlc::Parameter { " except in some cases where step is not an integer and" " floating point round-off affects the length of out. If the" " stop is equal to the start, the stop will be computed from" - " the output size"); + " the output shape."); DMLC_DECLARE_FIELD(step) .set_default(1) .describe("Spacing between values.");