在自定义Tensorflow单元中调用基本LSTM单元

时间:2017-05-08 18:46:11

标签: tensorflow

我试图从本文中实现MATCH LSTM:https://arxiv.org/pdf/1608.07905.pdf

我正在使用Tensorflow。该体系结构的一部分是RNN,其使用输入和先前状态来计算注意向量,其在将结果与输入连接并将其发送到LSTM之前应用于上下文。为了构建这个RNN的第一部分,我为Tensorflow编写了一个自定义单元来调用。但我不确定如何将结果发送到LSTM。是否可以在我写的自定义单元格中调用基本LSTM单元?我尝试了几种不同的方法,但不断收到错误"模块'对象没有属性' rnn_cell'"在调用LSTM单元的行。任何帮助将不胜感激!

编辑添加代码:

将numpy导入为np 导入tensorflow为tf

class MatchLSTMCell(tf.contrib.rnn.RNNCell):

PROCESSES

1 个答案:

答案 0 :(得分:1)

我也在尝试为实验重新实现Match_LSTM for Squad。 我使用MurtyShikhar's作为参考。有用!但是,他必须自定义AttentionWrapper并使用现有的BasicLSTM单元格。

我还尝试通过在Basic_LSTM中将z和state作为(输入,状态)对来创建​​Match_LSTM_cell:

inttypes.h