search

FL03 / acme

Acme aims to be a complete auto differentiation system written in Rust.
https://crates.io/crates/acme
Apache License 2.0
3 stars 0 forks source link

Generalize the implemented matrix multiplication method #71

Closed FL03 closed 5 months ago

FL03 commented 6 months ago

The current implementation only accounts for the two-dimensional case.

When both tensors are one-dimensional, a tensor of Rank(0) should be returned.