Skip to content

Fix matmul function in nn_transformer.py #1

Open
wmhst7 wants to merge 1 commit intodlsyscourse:mainfrom
wmhst7:main
Open

Fix matmul function in nn_transformer.py #1
wmhst7 wants to merge 1 commit intodlsyscourse:mainfrom
wmhst7:main

Conversation

@wmhst7
Copy link

@wmhst7 wmhst7 commented Apr 24, 2024

The origin implementation of matmul is not correct, which will lead to size mismatch errors. An updated implementation is given, through the same idea.

@wmhst7
Copy link
Author

wmhst7 commented Apr 24, 2024

Test cases passed after fix this issue:

image

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant

Comments