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

More mnist training. (#275)

Browse Source
This commit is contained in:
Laurent Mazare
2023-07-29 13:29:31 +01:00
committed by GitHub
parent c0a8ed19eb
commit 07eb899729
2 changed files with 130 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-nn/src/var_builder.rs
View File

@ -209,6 +209,7 @@ impl<'a> VarBuilder<'a> {
};
Ok(tensor)
}
pub fn get<S: Into<Shape>>(&self, s: S, tensor_name: &str) -> Result<Tensor> {
let data = self.data.as_ref();
let s: Shape = s.into();