我想修改tf.nn.rnn_cell.GRUCell
和另一门添加到它的第二输入,使得除了z
- 更新门和r
- 复位门会有第三g
- 定制的门第二个输入到网络同样Kiddon et al. 2016和Wen et al. 2015。
我设法为Torch7 GRU细胞做到这一点,它成功地工作了特定序列建模问题。我怀疑这是不是直接在Tensorflow实现,但也许我错了。可有人请提供Tensorflow定制细胞更多的细节?
有可能的。正如this文章,LSTM和GRU细胞说明仅仅是非线性和算术运算的安排。
This repository包含自定义LSTM,GRU和pyTorch其他RNN细胞实现。
This repository包含自定义LSTM和GRU实现了TensorFlow。