This is an automated email from the ASF dual-hosted git repository.
nswamy pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new 434a5fb move choose_element_0index to operator (#14273)
434a5fb is described below
commit 434a5fbd3103b1a01f10aa7d51a10d15d286f432
Author: Lai Wei <[email protected]>
AuthorDate: Mon Mar 4 09:53:38 2019 -0800
move choose_element_0index to operator (#14273)
---
src/ndarray/ndarray.cc | 5 -----
src/operator/tensor/broadcast_reduce_op_index.cc | 1 +
2 files changed, 1 insertion(+), 5 deletions(-)
diff --git a/src/ndarray/ndarray.cc b/src/ndarray/ndarray.cc
index b09d38a..3677127 100644
--- a/src/ndarray/ndarray.cc
+++ b/src/ndarray/ndarray.cc
@@ -2027,11 +2027,6 @@ MXNET_REGISTER_NDARRAY_FUN(_set_value)
MXNET_REGISTER_NDARRAY_FUN(_onehot_encode)
.set_function(BinaryOp<ndarray::OneHotEncode>);
-MXNET_REGISTER_NDARRAY_FUN(choose_element_0index)
-.set_function(BinaryOp<ndarray::MatChooseRowElem>)
-.describe("Choose one element from each line(row for python, column for
R/Julia)"
- " in lhs according to index indicated by rhs."
- " This function assume rhs uses 0-based index.");
MXNET_REGISTER_NDARRAY_FUN(fill_element_0index)
.set_function(TernaryOp<ndarray::MatFillRowElem>)
diff --git a/src/operator/tensor/broadcast_reduce_op_index.cc
b/src/operator/tensor/broadcast_reduce_op_index.cc
index ed9a90d..f3d1013 100644
--- a/src/operator/tensor/broadcast_reduce_op_index.cc
+++ b/src/operator/tensor/broadcast_reduce_op_index.cc
@@ -109,6 +109,7 @@ Examples::
.add_argument("data", "NDArray-or-Symbol", "The input array");
NNVM_REGISTER_OP(pick)
+.add_alias("choose_element_0index")
.describe(R"code(Picks elements from an input array according to the input
indices along the given axis.
Given an input array of shape ``(d0, d1)`` and indices of shape ``(i0,)``, the
result will be
