flip

paddle.fluid.layers.flip(input, dims, name=None):

该OP沿指定轴反转n维tensor.

参数

  • input (Variable) - 输入Tensor。维度为多维,数据类型为bool, int32, int64, float32或float64。

  • dims (list) - 需要翻转的轴。当 dims[i] < 0 时,实际的计算维度为 rank(input) + dims[i],其中i为dims的索引。

  • name (str|None) - 该参数供开发人员打印调试信息时使用,具体用法请参见 Name 。默认值为None。

返回

在指定dims上翻转后的Tensor,与输入input数据类型相同。

返回类型

Variable,与输入input数据类型相同。

抛出异常

  • TypeError - 当输出 out 和输入 input 数据类型不一致时候。

  • ValueError - 当参数 dims 不合法时。

代码示例 1

import paddle
import paddle.fluid as fluid
import numpy as np
input = fluid.data(name="x", shape=[-1, 2, 2], dtype='float32')
output = fluid.layers.nn.flip(input, dims=[0, 1])
exe = fluid.Executor(fluid.CPUPlace())
exe.run(fluid.default_startup_program())
img = np.arange(12).reshape((3,2,2)).astype(np.float32)
res = exe.run(fluid.default_main_program(), feed={'x':img}, fetch_list=[output])
print(res) # [[[10,11][8, 9]],[[6, 7],[4, 5]] [[2, 3],[0, 1]]]

代码示例 2

import paddle
import paddle.fluid as fluid
import numpy as np
img = np.arange(12).reshape((3,2,2)).astype(np.float32)
with fluid.dygraph.guard():
  inputs = fluid.dygraph.to_variable(img)
  ret = fluid.layers.flip(inputs, [0, 1])
  print(ret.numpy()) # [[[10,11][8, 9]],[[6, 7],[4, 5]] [[2, 3],[0, 1]]]