diff options
Diffstat (limited to 'candle-nn/src')
| -rw-r--r-- | candle-nn/src/var_builder.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-nn/src/var_builder.rs b/candle-nn/src/var_builder.rs index 5c222bf6..be1380b7 100644 --- a/candle-nn/src/var_builder.rs +++ b/candle-nn/src/var_builder.rs @@ -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(); |