【发布时间】:2016-10-21 03:27:25
【问题描述】:
当使用dynamic_rnn 时,需要一个名为initial_state 的参数。一个简单的解决方案是
initial_state = lstm_cell.zero_state(batch_size, tf.float32)
但是我想把初始状态设置为可以优化的参数,怎么办?
我可以定义两个 trainable_variables,分别称为 h0 和 c0,它们是两个向量。但是dynamic_rnn 需要两个矩阵,其中第一个维度是batch_size。如何将向量h0 扩展为每行为h0 的矩阵?
【问题讨论】:
标签: tensorflow recurrent-neural-network