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

来源:互联网 发布:软件测试学不会 编辑:程序博客网 时间:2024/06/05 20:52

如何初始化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 都为Tensorinitial_state = LSTMStateTuple(c_state, h_state)

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

1 0
原创粉丝点击