moveaxis

paddle. moveaxis ( x, source, destination, name=None ) [源代码]

将输入 Tensor x 的轴从 source 位置移动到 destination 位置,其他轴按原来顺序排布。同时根据新的 shape,重排 Tensor 中的数据。

参数

  • x (Tensor) - 输入的 N-D Tensor,数据类型为:bool、int32、int64、float32、float64、complex64、complex128。

  • source (int|tuple|list) - 将被移动的轴的位置,其每个元素必须为不同的整数。

  • destination (int|tuple|list) - 轴被移动后的目标位置,其每个元素必须为不同的整数。

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

Tensor:将轴移动后的 Tensor

代码示例

import paddle

x = paddle.ones([3, 2, 4])
paddle.moveaxis(x, [0, 1], [1, 2]).shape
# [4, 3, 2]

x = paddle.ones([2, 3])
paddle.moveaxis(x, 0, 1).shape # equivalent to paddle.t(x)
# [3, 2]