简单的一层lstm

”`python

使用 basic LSTM Cell.

lstm_cell = tf.contrib.rnn.BasicLSTMCell(n_hidden_units, forget_bias=1.0, state_is_tuple=True)
init_state = lstm_cell.zero_state(batch_size, dtype=tf.float32) # 初始化全零 state

如果使用tf.nn.dynamic_rnn(cell, inputs), 我们要确定 inputs 的格式. tf.nn.dynamic_rnn 中的 time_major 参数会针对不同 inputs 格式有不同的值.

如果 inputs 为 (batches, steps, inputs) ==> time_major=False;

如果 inputs 为 (steps, batches, inputs) ==> time_major=True;

outputs, final_state = tf.nn.dynamic_rnn(lstm_cell, X_in, initial_state=init_state, time_major=False)

outputs size: [batch_size, max_time, cell_state_size]
final_state[1] size: [batch_size, cell_state_size]
”`

Reference

各种LSTM代码比较全地址

分类导航

随笔深度学习报错解决效率工具 centos 算法题目数据结构未分类 python 网站书评 docker 书单 2019书单机器学习 linux python高阶教程 latex linux设备嵌入式信号与信息处理影评 django 科普爬虫自然语言处理 vue 运维系统优化数据库 git 视频短视频 Go html tensorflow 云加社区单片机驾考 kvm gnuradio markdown matlab 博客 mysql nextcloud vpn pandoc pyqt5 vscode 云服务网络代理 cmake javascript vim 网站日常