【发布时间】:2019-06-30 16:34:11
【问题描述】:
我想使用在this paper 中解释的减法门控 我用的是Tensorflow,目前代码是:(Using CPU)
import tensorflow.contrib.rnn as RNNCell
tgt_cell = RNNCell.LSTMCell(num_units=flags.hidden_size, state_is_tuple=True)
tgt_dropout_cell = RNNCell.DropoutWrapper(tgt_cell, output_keep_prob=self.keep_prob)
tgt_stacked_cell= RNNCell.MultiRNNCell([tgt_dropout_cell] * self.opt.num_layers, state_is_tuple=True)
根据论文的变化如下: 其中 LSTM 是:
门控应该是减法而不是乘法:
当我在我的代码中单击“LSTMCell”时,它会打开 rnn_cells.py,我不确定应该更改哪个部分。有人可以帮忙修改一下吗?
【问题讨论】:
标签: python tensorflow machine-learning neural-network lstm