deepjavalibrary / deepjavalibrary/djl
Add batch matrix-matrix product operation
Open
enhancement
- Dominant language
- Java
- Stars
- 4.9k
- Forks
- 759
- Avg merge
- 16h 15m
- Merged PRs (30d)
- 15
Description
## Description
Hello, there are a lot of functions from torch package not implemented (wrapped) in DJL.
For example batch matrix-matrix product operation: TORCH.BMM
It's looks like it should be wrapped in to NDArrays class.
May be there is some way i can dynamically create jni mapping for required function?
## References
https://pytorch.org/docs/stable/generated/torch.bmm.html#torch.bmm
Contributor guide
Assessment
This issue has not been assessed yet.