tensorflow源码分析——BasicLSTMCell

 

BasicLSTMCell 是最简单的LSTMCell，源码位于：/tensorflow/contrib/rnn/python/ops/core_rnn_cell_impl.py。
BasicLSTMCell 继承了RNNCell，源码位于：/tensorflow/python/ops/rnn_cell_impl.py

注意事项：
1. input_size 这个参数不能使用，使用的是num_units

2. state_is_tuple 官方建议设置为True。此时，输入和输出的states为c(cell状态)和h（输出）的二元组

3. 输入、输出、cell的维度相同，都是 batch_size * num_units，

cell = tf.contrib.rnn.BasicLSTMCell(num_units, forget_bias=0.0, state_is_tuple=True)　　#指定num_units
_initial_state = cell.zero_state(batch_size, tf.float32)　　　　　　　　　　　　　　　　　　　#指定batch_size,将c和h全部初始化为0，shape全是batch_size * num_units，