如何使用 Keras Lambda 层合并二维和转置

【问题标题】：How to use Keras Lambda layer to merge 2 dimensions and Transpose如何使用 Keras Lambda 层合并二维和转置
【发布时间】：2017-12-18 18:00:46
【问题描述】：

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

【问题讨论】：

【解决方案1】：

是的，您可以：

def reshapeTranspose(x):

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

图层：

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

PS：如果您使用的是 tensorflow，则不需要output_shape。
PS2：注意批量大小，通常它是 keras 张量中不是您自己创建的第一个维度。

【讨论】：