From b6f638f052589f103894b7560b239d1b013ce4b5 Mon Sep 17 00:00:00 2001 From: wanghaoshuang Date: Tue, 26 Dec 2017 10:38:24 +0800 Subject: [PATCH] Fix doc of transpose_op --- paddle/operators/transpose_op.cc | 34 +++++++++++++++++++++----------- 1 file changed, 23 insertions(+), 11 deletions(-) diff --git a/paddle/operators/transpose_op.cc b/paddle/operators/transpose_op.cc index f18be384345..7b3fa8062bf 100644 --- a/paddle/operators/transpose_op.cc +++ b/paddle/operators/transpose_op.cc @@ -72,17 +72,29 @@ Transpose Operator. The input tensor will be permuted according to the axis values given. The op functions is similar to how numpy.transpose works in python. -For example: input = numpy.arange(6).reshape((2,3)) -the input is: -array([[0, 1, 2], - [3, 4, 5]]) -given axis is: [1, 0] - -output = input.transpose(axis) -then the output is: -array([[0, 3], - [1, 4], - [2, 5]]) +For example: + + .. code-block:: python + + input = numpy.arange(6).reshape((2,3)) + + the input is: + + array([[0, 1, 2], + [3, 4, 5]]) + + given axis is: + + [1, 0] + + output = input.transpose(axis) + + then the output is: + + array([[0, 3], + [1, 4], + [2, 5]]) + So, given a input tensor of shape(N, C, H, W) and the axis is {0, 2, 3, 1}, the output tensor shape will be (N, H, W, C) -- GitLab