diff options
Diffstat (limited to 'candle-core/src/tensor.rs')
| -rw-r--r-- | candle-core/src/tensor.rs | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/candle-core/src/tensor.rs b/candle-core/src/tensor.rs index 9b0681e0..bcd380aa 100644 --- a/candle-core/src/tensor.rs +++ b/candle-core/src/tensor.rs @@ -444,6 +444,7 @@ impl Tensor { len, })? } + println!("Narrow {start:?} - {} - {len} - {dims:?}", dims[dim]); if start == 0 && dims[dim] == len { Ok(self.clone()) } else { @@ -452,10 +453,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, }; |