OliverFM / rust_light

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

Oliver/split out diagonal jvp utility #41

Closed OliverFM closed 1 year ago

OliverFM commented 1 year ago

This will make it much easier to deal with future element wise ops. It is not super efficient, since I have not baked broadcasting support into the utility, but that turned out to be quite hard to get right, so I will leave that for a later date.