例如:我有一个形状为 (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 后端。
最佳答案
使用 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)
关于tensorflow - 如何在 Keras 中翻转张量?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45014714/