Skip to content

Latest commit

 

History

History
25 lines (20 loc) · 822 Bytes

README.md

File metadata and controls

25 lines (20 loc) · 822 Bytes
Raw

An implementation of multiplicative LSTM in TensorFlow

This implementation is based on the following paper:

Ben Krause, Liang Lu, Iain Murray, and Steve Renals, "Multiplicative LSTM for sequence modelling, " in Workshop Track of ICLA 2017, https://openreview.net/forum?id=SJCS5rXFl&noteId=SJCS5rXFl

LICENSE

MIT license

Test

Tested on TensorFlow v1.1.0

Example usage:

import tensorflow as tf
from MultiplicativeLSTMCell import MultiplicativeLSTMCell

with tf.variable_scope("mlstm) as scope:
    lstm_size = 256
    lstm_init_value = tf.placeholder(tf.float32, shape=(None, 2*lstm_size))
    lstm_cell = MultiplicativeLSTMCell(self.lstm_size, forget_bias=1.0, state_is_tuple=False)
    lstm_outputs, lstm_new_state = tf.nn.dynamic_rnn(lstm_cell,xinput,dtype=tf.float32,initial_state=lstm_init_value)