带Attention机制的Seq2Seq框架梳理

本文详细梳理了带有Attention机制的Seq2Seq框架,包括Bi-RNN Encoder和Attention-Decoder的流程。在Encoder部分,使用双向GRU获取每个时刻的hidden_state。Decoder部分,通过Attention机制,结合Encoder输出和Decoder的prev_hidden_state生成能量,进一步计算概率并合成context向量。最终,使用全连接层将hidden_state映射为词汇表大小的向量,以生成目标词。
摘要由CSDN通过智能技术生成

借着与同事组内分享的机会,根据论文Neural Machine Translation By Jointly Learning to Align and Translate把带Attention机制的Seq2Seq框架Encoder与Decoder部分的流程图画了一下,公式梳理了一遍。

Bi-RNN Encoder

Bi-RNN Encoder

Encoder的流程如上图所示,最终的输出结果是每个时刻的hidden_state h1,h2,h3,...,hT

其中的GRU使用的双向的,正向部分的公式如下

h⃗ 0=0

z⃗ i=σ(W⃗ zE⃗ xi+U⃗ zh⃗ i1)(1)

r⃗ i=σ(W⃗ rE⃗ xi+U⃗ rh⃗ i1)(2)

h⃗ i^=tanh(W⃗ E⃗ xi+U⃗ [r⃗ i
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值