def model(self):
x1 = L.Input(shape=(None, ))
x2 = L.Input(shape=(self.hidden_size, ))
x3 = L.Input(shape=(self.hidden_size, ))
return tf.keras.Model(inputs=[x1, x2, x3],
outputs=self.call(x1, x2, x3),
name='AttGRUDecoder')
def model(self) -> tf.keras.Model:
x1 = L.Input(shape=(None,))
x2 = L.Input(shape=(self.hidden_size,))
return tf.keras.Model(inputs=[x1, x2],
outputs=self.call(x1, x2),
name='GRUEncoder')
