diff options
Diffstat (limited to 'candle-core/src')
| -rw-r--r-- | candle-core/src/error.rs | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/candle-core/src/error.rs b/candle-core/src/error.rs index f9e69122..30d06239 100644 --- a/candle-core/src/error.rs +++ b/candle-core/src/error.rs @@ -79,6 +79,13 @@ pub enum Error { nth_shape: Shape, }, + #[error("Cannot divide tensor of shape {shape:?} equally along dim {dim} into {n_parts}")] + ShapeMismatchSplit { + shape: Shape, + dim: usize, + n_parts: usize, + }, + #[error("{op} can only be performed on a single dimension")] OnlySingleDimension { op: &'static str, dims: Vec<usize> }, |