Skip to content

Commit 16d87f3

Browse files
weiyangfbjramseyer
authored andcommitted
Corrected doc in class RNNCell (pytorch#9866)
Summary: fixes pytorch#9642 Pull Request resolved: pytorch#9866 Differential Revision: D9012131 Pulled By: weiyangfb fbshipit-source-id: d2849b1a50234dbdb335dffab4835c9de85183c3
1 parent 3399147 commit 16d87f3

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
@@ -579,7 +579,7 @@ class RNNCell(RNNCellBase):
579579
580580
Attributes:
581581
weight_ih: the learnable input-hidden weights, of shape
582-
`(input_size x hidden_size)`
582+
`(hidden_size x input_size)`
583583
weight_hh: the learnable hidden-hidden weights, of shape
584584
`(hidden_size x hidden_size)`
585585
bias_ih: the learnable input-hidden bias, of shape `(hidden_size)`

0 commit comments

Comments
 (0)