初始状态的定义

来源:4-35 动手实现RNN-LSTM循环神经网络(十):编写测试方法

qq_windybeyondw_0

2018-11-20

状态(state)的存储和提取

    # 第二维是 2 是因为对每一个 LSTM 单元有两个来自上一单元的输入:
    # 一个是 前一时刻 LSTM 的输出 h(t-1)
    # 一个是 前一时刻的单元状态 C(t-1)
    # 这个 C 和 h 是用于构建之后的 tf.contrib.rnn.LSTMStateTuple
    self.init_state = tf.placeholder(tf.float32, [num_layers, 2, self.batch_size, self.hidden_size])
    老师这里定义初始状态为什么要写这四个参数呢?不是很懂
    我看其他代码有些定义初始状态都是用:
    initial_state = cell.zero_state(batch_size, tf.float32)
    直接定义了初始状态,也没有再去定义每一层的状态之类,请问这两者有什么区别呢?
写回答

1回答

Oscar

2018-11-21

不同的初始方法吧。我的是基于 PPT 里的 LSTM 原理图

0
0

基于Python玩转人工智能最火框架 TensorFlow应用实践

机器学习入门,打牢TensorFlow框架应用是关键!

2214 学习 · 688 问题

查看课程