This is an automated email from the ASF dual-hosted git repository.

haibin pushed a commit to branch fix-doc
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git

commit 551233f77e61d64f16be43e7a1d384d9b12c6e7e
Author: Haibin Lin <[email protected]>
AuthorDate: Sun Aug 4 21:23:13 2019 -0700

    Update init_op.cc
---
 src/operator/tensor/init_op.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/operator/tensor/init_op.cc b/src/operator/tensor/init_op.cc
index 0cbdaa4..239cff7 100644
--- a/src/operator/tensor/init_op.cc
+++ b/src/operator/tensor/init_op.cc
@@ -134,7 +134,8 @@ Examples::
     [](const NodeAttrs& attrs) { return std::vector<uint32_t>(1, 0); })
 .set_attr<FCompute>("FCompute<cpu>", RangeCompute<cpu, RangeLikeParam>)
 .set_attr<nnvm::FGradient>("FGradient", MakeZeroGradNodes)
-.add_argument("data", "NDArray-or-Symbol", "The input");
+.add_argument("data", "NDArray-or-Symbol", "The input")
+.add_arguments(RangeLikeParam::__FIELDS__());
 
 NNVM_REGISTER_OP(_linspace)
 .describe("Return evenly spaced numbers over a specified interval. Similar to 
Numpy")

Reply via email to