-
Notifications
You must be signed in to change notification settings - Fork 103
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Added Dot Layer #275
Added Dot Layer #275
Conversation
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/merge/Dot.kt
Show resolved
Hide resolved
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/merge/Dot.kt
Outdated
Show resolved
Hide resolved
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/merge/Dot.kt
Outdated
Show resolved
Hide resolved
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/merge/Dot.kt
Outdated
Show resolved
Hide resolved
Hi @therealansh do you have plans to fix a review comments here? It will be great to merge it in December in the codebase |
Added the docs! I have removed the Dot Test case for now. I'll be doing it in the PR for #276. Should i add a TODO as well for refactoing l2Normalize and batchDot for future refactoring? |
No needs to add TODO, I make this refactoring in the nearest future |
Great, now we have DOT! |
Hey @zaleslaw , while writing test cases for
at out = MatMul.create(scope, x2, y2, MatMul.transposeA(adjX), MatMul.transposeB(adjY)) in batchDot. Can you review it once I cannot understand what the issue is about? |
This PR adds Dot layer and closes #144. @zaleslaw would you mind reviewing it for this initial commit. I am still working on some test cases.