Skip to content

Conversation

@rcurtin
Copy link
Member

@rcurtin rcurtin commented Mar 31, 2025

As @nzachary pointed out in #3912, currently, the LSTM layer computes the size of its input by accumulating the input dimensions. But this is not correct: it should be the product instead.

This simple PR fixes that and adds a test case.

Copy link
Contributor

@eddelbuettel eddelbuettel left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

👍

@rcurtin rcurtin merged commit 5a14c25 into mlpack:master Mar 31, 2025
15 checks passed
@rcurtin rcurtin deleted the lstm-input-size-fix branch March 31, 2025 19:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants