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

Cosmetic changes.

Browse Source
This commit is contained in:
laurent
2023-06-19 21:30:03 +01:00
parent 26d6288eb6
commit bcae61b7f2
3 changed files with 33 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/shape.rs
View File

@ -1,6 +1,12 @@
use crate::{Error, Result};
pub struct Shape(pub(crate) Vec<usize>);
impl std::fmt::Debug for Shape {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "{:?}", &self.dims())
}
}
impl From<&[usize; 1]> for Shape {
fn from(dims: &[usize; 1]) -> Self {
Self(dims.to_vec())