2016-09-28 65 views
1

在張量流LSTM單元格中,我們只通過單元的數量(num_units)來初始化它,這是單元格的隱藏大小。但是,該單元也應該輸入x。由於基礎操作是矩陣乘法運算,所以x的大小也應該在設置LSTM權重中起作用。那麼,單元格是否應該將輸入x的大小也作爲初始化參數?tensorlow LSTM爲什麼只將隱藏大小作爲輸入?

回答

1

LSTM細胞(或其他類型的RNN細胞)只是您傳遞給函數的對象,如tf.nn.rnntf.nn.rnn需要輸入單元格和數據。使用您傳遞給tf.nn.rnn的數據在rnn函數中調用單元格。