Add arange_like to npx (apache#16883)

yajiedesign · Nov 23, 2019 · 971ea4f · 971ea4f
1 parent 6c20fb9
commit 971ea4f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/operator/tensor/init_op.cc b/src/operator/tensor/init_op.cc
@@ -103,6 +103,7 @@ NNVM_REGISTER_OP(_arange)
 .add_arguments(RangeParam::__FIELDS__());
 
 NNVM_REGISTER_OP(_contrib_arange_like)
+.add_alias("_npx_arange_like")
 .describe(R"code(Return an array with evenly spaced values. If axis is not given, the output will 
 have the same shape as the input array. Otherwise, the output will be a 1-D array with size of 
 the specified axis in input shape.