如何使用Keras Lambda图层合并2个维度和Transpose

Question

我有一个形状矢量（1,9,128）。我需要它转换为（128,9），但不是简单的重塑。我希望1和9合并然后转置。如何使用keras中的Lambda图层执行此操作？谢谢

Answer 1

是的你可以：

def reshapeTranspose(x):

    x = K.reshape(x,(9,128))
    return K.permute_dimensions(x,(1,0))

图层：

Lambda(reshapeTranspose, output_shape=(128,9))

PS：如果你使用张量流，则不需要output_shape。 PS2：小心批量大小，通常它是你自己没有创建的keras张量的第一个维度。