huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Start adding some ops.

Browse Source
This commit is contained in:
laurent
2023-06-20 08:41:19 +01:00
parent ef6760117f
commit 7a31ba93e4
5 changed files with 83 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/shape.rs
View File

@ -1,4 +1,6 @@
use crate::{Error, Result};
#[derive(Clone, PartialEq, Eq)]
pub struct Shape(pub(crate) Vec<usize>);
impl std::fmt::Debug for Shape {
@ -56,6 +58,10 @@ impl From<(usize, usize, usize)> for Shape {
}
impl Shape {
pub fn from_dims(dims: &[usize]) -> Self {
Self(dims.to_vec())
}
pub fn rank(&self) -> usize {
self.0.len()
}
@ -76,7 +82,7 @@ impl Shape {
Err(Error::UnexpectedNumberOfDims {
expected: 0,
got: shape.len(),
shape: shape.to_vec(),
shape: self.clone(),
})
}
}
@ -89,7 +95,7 @@ impl Shape {
Err(Error::UnexpectedNumberOfDims {
expected: 1,
got: shape.len(),
shape: shape.to_vec(),
shape: self.clone(),
})
}
}
@ -102,7 +108,7 @@ impl Shape {
Err(Error::UnexpectedNumberOfDims {
expected: 2,
got: shape.len(),
shape: shape.to_vec(),
shape: self.clone(),
})
}
}
@ -115,7 +121,7 @@ impl Shape {
Err(Error::UnexpectedNumberOfDims {
expected: 3,
got: shape.len(),
shape: shape.to_vec(),
shape: self.clone(),
})
}
}
@ -128,7 +134,7 @@ impl Shape {
Err(Error::UnexpectedNumberOfDims {
expected: 4,
got: shape.len(),
shape: shape.to_vec(),
shape: self.clone(),
})
}
}