huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 03:28:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move the variable creation to the variable module. (#159)

Browse Source 
* Move the variable creation to the variable module.

* Make it possible to set a variable.

* Add some basic gradient descent test.

* Get the gradient descent test to work.
This commit is contained in:
Laurent Mazare
2023-07-13 16:55:40 +01:00
committed by GitHub
parent 6991036bc5
commit 5ee3c95582
5 changed files with 137 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/src/lib.rs
View File

@ -68,7 +68,7 @@ pub use shape::{Shape, D};
pub use storage::Storage;
use strided_index::StridedIndex;
pub use tensor::{Tensor, TensorId};
pub use variable::Variable;
pub use variable::Var;
#[cfg(feature = "cuda")]
pub use cuda_backend::{CudaDevice, CudaStorage};