OliverFM / rust_light

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

Generalise Autograd to work over RcTensor and TensorViews #33

Open OliverFM opened 1 year ago

OliverFM commented 1 year ago

To do this I need to make TensorViews wrapped by an RC, and make some trait to generalise over the two.

This will be pretty important for calculating jacobians -- since I will want to flatten everything before doing the chain rule, but need to make sure that the same underlying tensor has the grad set.