【发布时间】:2016-07-21 01:35:38
【问题描述】:
我想使用 TensorFlow 训练一个 RNN 语言模型。
我的训练数据是一个由 5 个标记组成的序列,用整数表示
x = [0, 1, 2, 3, 4]
我希望 RNN 的展开长度为 4,训练批次大小为 2。(我选择这些值是为了需要填充。)
每个令牌都有一个长度为 3 的嵌入
0 -> [0, 0 ,0]
1 -> [10, 10, 10]
2 -> [20, 20, 20]
3 -> [30, 30, 30]
4 -> [40, 40, 40]
我应该将什么作为参数传递给tf.nn.dynamic_rnn?
这主要是"How is the input tensor for TensorFlow's tf.nn.dynamic_rnn operator structured?" 的转贴。 Eugene Brevdo 对此很有帮助。但是他稍微误解了我的问题,因为我没有足够的 TensorFlow 知识来清楚地提出这个问题。 (具体来说,他认为我的意思是批量大小为 1。)与其通过编辑原始问题来冒额外的混淆,我认为如果我在这里重新表述它是最清楚的。
我正在尝试通过写Example TensorFlow RNN Language Model 为自己解决这个问题。
【问题讨论】:
标签: tensorflow recurrent-neural-network