如何在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后端，可以使用该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)