未验证 提交 a0e88941 编写于 作者: W whs 提交者: GitHub

Merge pull request #7020 from wanghaoshuang/fix_transpose_doc

Fix transpose doc
...@@ -72,17 +72,29 @@ Transpose Operator. ...@@ -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:
the input is:
array([[0, 1, 2], .. code-block:: text
[3, 4, 5]])
given axis is: [1, 0] input = numpy.arange(6).reshape((2,3))
output = input.transpose(axis) the input is:
then the output is:
array([[0, 3], array([[0, 1, 2],
[1, 4], [3, 4, 5]])
[2, 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}, 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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册