flip¶
沿指定轴反转 n 维 Tensor。
参数¶
x (Tensor) - 输入的 Tensor。维度为多维,数据类型为 bool、int32、int64、float32 或 float64。
axis (list|tuple|int) - 需要翻转的轴。当 axis 是 int 型时,且 axis<0,实际的计算维度为 ndim(x) + axis;axis 是 list 或 tuple 时,
axis[i] < 0
时,实际的计算维度为 ndim(x) + axis[i],其中 i 为 axis 的索引。name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。
返回¶
Tensor,在指定 axis 上翻转后的 Tensor,与输入 x 数据类型相同。
代码示例¶
>>> import paddle
>>> image_shape=(3, 2, 2)
>>> img = paddle.arange(image_shape[0] * image_shape[1] * image_shape[2]).reshape(image_shape)
>>> tmp = paddle.flip(img, [0,1])
>>> print(tmp)
Tensor(shape=[3, 2, 2], dtype=int64, place=Place(cpu), stop_gradient=True,
[[[10, 11],
[8 , 9 ]],
[[6 , 7 ],
[4 , 5 ]],
[[2 , 3 ],
[0 , 1 ]]])
>>> out = paddle.flip(tmp,-1)
>>> print(out)
Tensor(shape=[3, 2, 2], dtype=int64, place=Place(cpu), stop_gradient=True,
[[[11, 10],
[9 , 8 ]],
[[7 , 6 ],
[5 , 4 ]],
[[3 , 2 ],
[1 , 0 ]]])