是否可以创建自定义Tensorflow GRU / LSTM细胞?

问题描述 投票:3回答:1

我想修改tf.nn.rnn_cell.GRUCell和另一门添加到它的第二输入,使得除了z - 更新门和r - 复位门会有第三g - 定制的门第二个输入到网络同样Kiddon et al. 2016Wen et al. 2015

enter image description here

我设法为Torch7 GRU细胞做到这一点,它成功地工作了特定序列建模问题。我怀疑这是不是直接在Tensorflow实现,但也许我错了。可有人请提供Tensorflow定制细胞更多的细节?

tensorflow
1个回答
1
投票

有可能的。正如this文章,LSTM和GRU细胞说明仅仅是非线性和算术运算的安排。

This repository包含自定义LSTM,GRU和pyTorch其他RNN细胞实现。

This repository包含自定义LSTM和GRU实现了TensorFlow。

© www.soinside.com 2019 - 2024. All rights reserved.