Wie man Keras Lambda-Schicht verwendet, um 2 Dimensionen zu verschmelzen und zu transponieren

Question

Ich habe einen Vektor der Form (1,9,128). Ich brauche es konvertiert zu (128,9), aber nicht mit einer einfachen Umformung. Ich möchte die 1 und 9 zusammenführen und dann transponieren. Wie kann ich das mit der Lambda-Schicht in Keras machen? Danke

Answer 1

Ja, Sie können:

def reshapeTranspose(x): 

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

Die Schicht:

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

PS: output_shape ist nicht notwendig, wenn Sie tensorflow verwenden.
PS2: Vorsicht vor der Losgröße, normalerweise ist es die erste Dimension in Keras Tensoren, die Sie nicht selbst erstellt haben.