tensorflow学习笔记(三十七):如何自定义LSTM的initial state

如何初始化LSTM的state

LSTM 需要 initial state。一般情况下,我们都会使用 lstm_cell.zero_state()来获取 initial state。但有些时候,我们想要给 lstm_cellinitial state 赋予我们想要的值,而不是简单的用 0 来初始化,那么,应该怎么做呢? 当然,当我们设置了state_is_tuple=False的时候,是很简单的,当state_is_tuple=True的时候,应该怎么做呢? 需要用到LSTMStateTuple

LSTMStateTuple(c ,h)

可以把 LSTMStateTuple() 看做一个op

from tensorflow.contrib.rnn.python.ops.core_rnn_cell_impl import LSTMStateTuple

...
c_state = ...
h_state = ...
# c_state , h_state 都为Tensor
initial_state = LSTMStateTuple(c_state, h_state)

当然,GRU就没有这么麻烦了,因为GRU没有两个state

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏简书专栏

房价数据转换和清洗

下载链接:https://pan.baidu.com/s/16D5hw-XBEQnwtsf4fDJ8xw 密码:e1fg

1042
来自专栏机器之心

业界 | Facebook 开源语音识别工具包wav2letter(附实现教程)

4138
来自专栏和蔼的张星的图像处理专栏

LCT代码跑起来先文章思路总结

论文才刚开始看,但是代码先跑了一下看结果,有一点小坑,记录下: 首先去论文的github上去下载代码:点这里 readme里其实写了怎么搞:

1423
来自专栏机器学习人工学weekly

机器学习人工学weekly-2018/7/22

链接:https://cn.udacity.com/course/self-driving-car-fundamentals-featuring-apollo-...

591
来自专栏数据结构与算法

9.21模拟赛解题报告

上来看T1,咦?我好像做过这题在仙人掌上的版本。。树上更简单吧。。写+拍 1h,期间拍出了暴力的两个bug。。。

684
来自专栏数说工作室

创业板、市盈率、Python!|【量化小讲堂】计算创业板平均市盈率

数说君的文前话 本文开始正式进入python的金融数据学习,为更好的学习,数说君为大家准备了一些基础知识。 → 如果对python完全不了解,点击这里: 统计师...

3414
来自专栏专知

TensorFlow 1.9 新增 tf.keras 官方入门教程(Keras与TF的深度集成)

【导读】Keras是深度学习框架中最适合工程化的框架之一,具有简单友好的接口。TensorFlow虽然功能强大,但是对于工程师来说,它的使用却十分的繁琐。好消息...

4692
来自专栏专知

【干货】还在自己写训练过程么?你需要一个训练引擎

2793
来自专栏安富莱嵌入式技术分享

【安富莱二代示波器教程】第10章 示波器设计—数字信号处理

本章节为大家讲解二代示波器中用到的FFT和FIR。单纯从应用上来说,比较省事,调用API函数即可,从学习的角度来说,需要大家花点精力。

583
来自专栏码字搬砖

kylin报错 java.lang.ArrayIndexOutOfBoundsException: -1

当用kylin做报表时,kylin报错 2017-06-26 14:51:52,523 ERROR [IPC Server handler 3 on 330...

1064

扫码关注云+社区