Skip to content

Commit e9bf702

Browse files
makarandtapaswiapaszke
authored andcommitted
LSTM bias_hh, fix docs
Rename W_hi ... to b_hi ...
1 parent 3716286 commit e9bf702

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
@@ -259,7 +259,7 @@ class LSTM(RNNBase):
259259
`(hidden_size x 4*hidden_size)`
260260
bias_ih_l[k] : the learnable input-hidden bias of the k-th layer `(b_ii|b_if|b_ig|b_io)`, of shape
261261
`(4*hidden_size)`
262-
bias_hh_l[k] : the learnable hidden-hidden bias of the k-th layer `(W_hi|W_hf|W_hg|b_ho)`, of shape
262+
bias_hh_l[k] : the learnable hidden-hidden bias of the k-th layer `(b_hi|b_hf|b_hg|b_ho)`, of shape
263263
`(4*hidden_size)`
264264
265265
Examples::

0 commit comments

Comments
 (0)