OliverFM / rust_light

A tensor processing library in rust.
MIT License
3 stars 0 forks source link

Oliver/small bmm bug fixes #43

Closed OliverFM closed 1 year ago

OliverFM commented 1 year ago

This gets the linear layer a little better -- this converts forward to have the same matrix multiplication order as torch's linear layer