Merge pull request #127 from LaurentMazare/tensor_indexing

`i(..)` indexing sugar (partial).
2025-06-18 11:37:11 +00:00 · 2023-07-10 19:56:34 +02:00
parent f29b77ec19 1ad235953b
commit fba07d6b6b
3 changed files with 246 additions and 1 deletions
--- a/candle-core/src/tensor.rs
+++ b/candle-core/src/tensor.rs
@ -504,10 +504,11 @@ impl Tensor {
            } else {
                None
            };
+            let layout = self.layout().narrow(dim, start, len)?;
            let tensor_ = Tensor_ {
                id: TensorId::new(),
                storage: self.storage.clone(),
-                layout: self.layout().narrow(dim, start, len)?,
+                layout,
                op,
                is_variable: false,
            };