We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0c69fd5 commit 017c7efCopy full SHA for 017c7ef
torch/nn/modules/rnn.py
@@ -374,7 +374,7 @@ class LSTMCell(RNNCellBase):
374
Inputs: `input, (h_0, c_0)`
375
- `input` : A `(batch x input_size)` tensor containing input features
376
- `h_0` : A `(batch x hidden_size)` tensor containing the initial hidden state for each element in the batch.
377
- - `c_0` : A `(batch x hidden_size)` tensor containing the initial hidden state for each element in the batch.
+ - `c_0` : A `(batch x hidden_size)` tensor containing the initial cell state for each element in the batch.
378
379
Outputs: `h_1, c_1`
380
- h_1: A `(batch x hidden_size)` tensor containing the next hidden state for each element in the batch
0 commit comments