Skip to content

RNNCell doc on weight_ih has the shape dimension reversed #9642

@cuimenc

Description

@cuimenc

Issue description

RNNCell doc on weight_ih has the shape dimension reversed

Doc says:

weight_ih: the learnable input-hidden weights, of shape (input_size x hidden_size)

But the actual shape is (hidden_size x input_size)

self.weight_ih = Parameter(torch.Tensor(hidden_size, input_size))

Metadata

Metadata

Assignees

Labels

module: docsRelated to our documentation, both in docs/ and docblockstodoNot as important as medium or high priority tasks, but we will work on these.

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions