gorgonia / gorgonia

Gorgonia is a library that helps facilitate machine learning in Go.
https://gorgonia.org/
Apache License 2.0
5.4k stars 431 forks source link

Fix BatchedMatMul error when operating on 1x1 matrix #553

Closed wzzhu closed 9 months ago

wzzhu commented 9 months ago

Fix https://github.com/gorgonia/gorgonia/issues/552