Skip to content

Commit 017c7ef

Browse files
TomVeniatapaszke
authored andcommitted
Fix typo in LSTMCell documentation
1 parent 0c69fd5 commit 017c7ef

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/nn/modules/rnn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -374,7 +374,7 @@ class LSTMCell(RNNCellBase):
374374
Inputs: `input, (h_0, c_0)`
375375
- `input` : A `(batch x input_size)` tensor containing input features
376376
- `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.
377+
- `c_0` : A `(batch x hidden_size)` tensor containing the initial cell state for each element in the batch.
378378
379379
Outputs: `h_1, c_1`
380380
- h_1: A `(batch x hidden_size)` tensor containing the next hidden state for each element in the batch

0 commit comments

Comments
 (0)