Merge pull request #7020 from wanghaoshuang/fix_transpose_doc

Fix transpose doc
del_some_in_makelist
whs 7 years ago committed by GitHub
commit a0e8894137
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -72,17 +72,29 @@ Transpose Operator.
The input tensor will be permuted according to the axis values given. The input tensor will be permuted according to the axis values given.
The op functions is similar to how numpy.transpose works in python. The op functions is similar to how numpy.transpose works in python.
For example: input = numpy.arange(6).reshape((2,3)) For example:
.. code-block:: text
input = numpy.arange(6).reshape((2,3))
the input is: the input is:
array([[0, 1, 2], array([[0, 1, 2],
[3, 4, 5]]) [3, 4, 5]])
given axis is: [1, 0]
given axis is:
[1, 0]
output = input.transpose(axis) output = input.transpose(axis)
then the output is: then the output is:
array([[0, 3], array([[0, 3],
[1, 4], [1, 4],
[2, 5]]) [2, 5]])
So, given a input tensor of shape(N, C, H, W) and the axis is {0, 2, 3, 1}, 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) the output tensor shape will be (N, H, W, C)

Loading…
Cancel
Save