flip¶
- paddle.flip(x, axis, name=None):
该OP沿指定轴反转n维tensor.
参数¶
x (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|None) - 该参数供开发人员打印调试信息时使用,具体用法请参见 Name 。默认值为None。
返回¶
Tensor,在指定axis上翻转后的张量,与输入x数据类型相同。
代码示例¶
import paddle
import numpy as np
image_shape=(3, 2, 2)
x = np.arange(image_shape[0] * image_shape[1] * image_shape[2]).reshape(image_shape)
x = x.astype('float32')
img = paddle.to_tensor(x)
out = paddle.flip(img, [0,1])
print(out) # [[[10,11], [8, 9]], [[6, 7], [4, 5]], [[2, 3],[0, 1]]]