deepjavalibrary / deepjavalibrary/djl

Add batch matrix-matrix product operation

Open
#2,349 5 comments 0 reactions 0 assignees View on GitHub
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.