3D-Tensor-Eingang zum Einbetten der Schicht in Keras oder Tensorflow?

Question

Ich möchte ein Netzwerk aufbauen, das Sätze als Eingabe zur Vorhersage der Stimmung verwendet. So sieht meine Eingabe ungefähr so ​​aus (Anzahl der Stichproben x Anzahl der Sätze x Anzahl der Wörter). Ich möchte es dann in eine Einbettungsschicht einspeisen, um die Wortvektoren zu lernen, die dann summiert werden können, um einen Satzvektor zu erhalten. Ist diese Art von Architektur in Keras möglich? oder Tensorflow? Aus der Dokumentation nimmt die Einbettungsschicht von Keras nur Eingaben (nb_samples, sequence_length) auf. Ist eine Arbeit möglich?

Answer 1

Ich denke, diese Klasse für Keras löst:

class AnyShapeEmbedding(Embedding): 
    ''' 
    This Embedding works with inputs of any number of dimensions. 
    This can be accomplished by simply changing the output shape computation. 
    ''' 
    #@overrides 
    def compute_output_shape(self, input_shape): 
     return input_shape + (self.output_dim,)