Aviral Garg 27d52d6378 Fix MHAEinsum weight dimension bug when d_in != d_out (#857) (#893) ...

* Fix MHAEinsum weight dimension bug when d_in != d_out (#857)

Previously MHAEinsum initialized weight matrices with shape (d_out, d_in) and used inappropriate einsum notation, causing failures for non-square input-output dimensions. This commit corrects weight initialization to shape (d_in, d_out), updates einsum notation to 'bnd,do->bno', and adds three unit tests to verify parity across different d_in and d_out settings. All tests pass successfully.

* use pytest

* Update .gitignore

---------

Co-authored-by: rasbt <mail@sebastianraschka.com>

2025-10-31 21:45:31 -05:00

..

ISSUE_TEMPLATE

minor fixes (#248)

2024-06-26 08:52:51 -05:00

scripts

Make quote style consistent (#891)

2025-10-21 19:42:33 -05:00

workflows

Fix MHAEinsum weight dimension bug when d_in != d_out (#857) (#893)

2025-10-31 21:45:31 -05:00