huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 03:54:56 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
52c5d8c087f6a2ee91b807467860eb3e96bb6267
candle/candle-examples/examples/simple-training
History
Laurent Mazare 52c5d8c087 Add the gather op. (#219) 
* Start adding gather.

* Gather cpu implementation + use in simple training.

* Add scatter_add for the gradient of gather.

* Simple cpu implementation of scatter_add.

* Use gather in the simple-training backprop.
2023-07-22 07:21:28 +01:00
..
main.rs
Add the gather op. (#219)
2023-07-22 07:21:28 +01:00