From 3165262f1f4c94aaf9bb9a333f32d4c94fddf293 Mon Sep 17 00:00:00 2001 From: Tao Lv Date: Tue, 22 Jan 2019 05:50:32 +0800 Subject: [PATCH] fix doc of take operator (#13947) --- src/operator/tensor/indexing_op.cc | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/src/operator/tensor/indexing_op.cc b/src/operator/tensor/indexing_op.cc index c39418dbe41d..564171d2c3fd 100644 --- a/src/operator/tensor/indexing_op.cc +++ b/src/operator/tensor/indexing_op.cc @@ -678,11 +678,14 @@ Examples:: // In this case we will get rows 0 and 1, then 1 and 2 (calculated by wrapping around). // Along axis 1 - take(x, [[0, 3], [-1, -2]], axis=1, mode='wrap') = [[[ 1., 2.], - [ 3., 4.]], + take(x, [[0, 3], [-1, -2]], axis=1, mode='wrap') = [[[ 1. 2.] + [ 2. 1.]] - [[ 3., 4.], - [ 5., 6.]]] + [[ 3. 4.] + [ 4. 3.]] + + [[ 5. 6.] + [ 6. 5.]]] The storage type of ``take`` output depends upon the input storage type: