Keras Lambdaレイヤーを使用して2次元をマージしてTransposeを使う方法

Question

私は形状のベクトル（1,9,128）を持っています。私はそれを（128,9）に変換する必要がありますが、単純な形では変換できません。 1と9をマージしてから転置したい。ケラスのラムダレイヤーでどうすればいいですか？ありがとう

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：あなたはtensorflowを使用している場合output_shapeは必要ありません。
PS2：バッチサイズには注意してください。通常は、自分で作成していないケラステンソルの最初の次元です。