tensorflow - 如何在 Keras 中翻转张量？

Question

例如:我有一个形状为 (5,10) 的张量我想要一个形状为 (5,10) 的张量但第一个元素现在应该是最后一个元素。所以[1,2,3,4,5]变成 [5,4,3,2,1]和 [[1,2,3,4,5],[2,3,4,5,6]]变成 [[2,3,4,5,6],[1,2,3,4,5]] .

如果重要的话，我正在使用 tensorflow 后端。

Answer 1

使用 Keras backend ，还有reverse功能。

import keras.backend as K

flipped = K.reverse(x,axes=0)

要在层中使用它，您可以创建一个 Lambda 层:

from keras.layers import *
layer = Lambda(lambda x: K.reverse(x,axes=0),output_shape=(shape of x))

(如果是顺序层， model.add(layer) ，如果是函数式 API 模型， output = layer(input)