如何在Keras中翻转Tensor？

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)