在张量流LSTM单元格中,我们只按单元格数(num_units
)初始化它,这是单元格的隐藏大小。但是,单元格也应该输入x
。并且由于基础操作是矩阵乘法运算,x
的大小也应该在设置LSTM权重中起作用。那么,单元格不应该将输入x
的大小作为初始化参数吗?
答案 0 :(得分:1)
LSTM单元格(或其他类型的RNN单元格)只是您传递给函数的对象,例如tf.nn.rnn
。 tf.nn.rnn
获取输入的单元格和数据。使用您传递给tf.nn.rnn
的数据在rnn函数内部调用Cell。