提交 b6f638f0 编写于 作者: W wanghaoshuang

Fix doc of transpose_op

上级 682c184c
...@@ -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:: python
input = numpy.arange(6).reshape((2,3))
the input is:
array([[0, 1, 2],
[3, 4, 5]]) [3, 4, 5]])
given axis is: [1, 0]
output = input.transpose(axis) given axis is:
then the output is:
array([[0, 3], [1, 0]
output = input.transpose(axis)
then the output is:
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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册