From 48cf48b2224b6afd42028b475d37f689ed061ac5 Mon Sep 17 00:00:00 2001 From: Siva <sivar.b@huawei.com> Date: Mon, 24 Sep 2018 22:57:12 +0530 Subject: [PATCH] [DOC] Argument name correction. (#1765) --- nnvm/src/top/tensor/transform.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nnvm/src/top/tensor/transform.cc b/nnvm/src/top/tensor/transform.cc index b1485438c..40c8c930a 100644 --- a/nnvm/src/top/tensor/transform.cc +++ b/nnvm/src/top/tensor/transform.cc @@ -171,7 +171,7 @@ Example:: y = [[3,3],[4,4],[5,5]] z = [[6,6], [7,7],[8,8]] - concatenate(x,y,z,dim=0) = [[ 1., 1.], + concatenate(x,y,z,axis=0) = [[ 1., 1.], [ 2., 2.], [ 3., 3.], [ 4., 4.], @@ -183,7 +183,7 @@ Example:: Note that you cannot concat x,y,z along dimension 1 since dimension 0 is not the same for all the input arrays. - concatenate(y,z,dim=1) = [[ 3., 3., 6., 6.], + concatenate(y,z,axis=1) = [[ 3., 3., 6., 6.], [ 4., 4., 7., 7.], [ 5., 5., 8., 8.]] -- GitLab