diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-12-22 09:18:13 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-12-22 09:18:13 +0100 |
| commit | 62ced44ea94da7062430ed6c21ff17b36f41737d (patch) | |
| tree | ffcb633955da0d743b013266de9b8b45bd59a1f0 /candle-core/src/tensor_cat.rs | |
| parent | 5c2f893e5aa21c9f7c82a00407edb6d76db1d06c (diff) | |
| download | candle-62ced44ea94da7062430ed6c21ff17b36f41737d.tar.gz candle-62ced44ea94da7062430ed6c21ff17b36f41737d.tar.bz2 candle-62ced44ea94da7062430ed6c21ff17b36f41737d.zip | |
Add a Context trait similar to anyhow::Context. (#2676)
* Add a Context trait similar to anyhow::Context.
* Switch two unwrap to context.
Diffstat (limited to 'candle-core/src/tensor_cat.rs')
| -rw-r--r-- | candle-core/src/tensor_cat.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-core/src/tensor_cat.rs b/candle-core/src/tensor_cat.rs index 204e7fd6..be6dfe61 100644 --- a/candle-core/src/tensor_cat.rs +++ b/candle-core/src/tensor_cat.rs @@ -1,4 +1,4 @@ -use crate::{shape::Dim, Error, Result, Shape, Tensor}; +use crate::{shape::Dim, Context, Error, Result, Shape, Tensor}; impl Tensor { /// Concatenates two or more tensors along a particular dimension. @@ -134,7 +134,7 @@ impl Tensor { .bt())? } } - let next_offset = offsets.last().unwrap() + arg.elem_count(); + let next_offset = offsets.last().context("empty offsets")? + arg.elem_count(); offsets.push(next_offset); } let shape = Shape::from(cat_dims); |