小编Jas*_*hye的帖子

如何使用tensorflow注意力层?

我试图了解如何使用tf.keras.layers.Attention此处显示的内容:

张量流注意力层

我正在尝试将它与编码器解码器 seq2seq 模型一起使用。下面是我的代码:

encoder_inputs = Input(shape=(max_len_text,)) 
enc_emb = Embedding(x_voc_size, latent_dim,trainable=True)(encoder_inputs) 
encoder_lstm=LSTM(latent_dim, return_state=True, return_sequences=True) 
encoder_outputs, state_h, state_c= encoder_lstm(enc_emb) 

decoder_inputs = Input(shape=(max_len_summary,)) 
dec_emb_layer = Embedding(y_voc_size, latent_dim,trainable=True) 
dec_emb = dec_emb_layer(decoder_inputs) 

decoder_lstm = LSTM(latent_dim, return_sequences=True, return_state=True) 
decoder_outputs,decoder_fwd_state, decoder_back_state = decoder_lstm(dec_emb,initial_state=[state_h, state_c]) 
Run Code Online (Sandbox Code Playgroud)

我的问题是,如何在这个模型中使用 keras 中给定的注意力层?我无法理解他们的文件。

python keras tensorflow seq2seq

5
推荐指数
1
解决办法
2125
查看次数

标签 统计

keras ×1

python ×1

seq2seq ×1

tensorflow ×1