Skip to main content

ml_tensor_basic_usage/
basic_usage.rs

1use use_ml_tensor::{TensorDType, TensorShape};
2
3fn main() -> Result<(), use_ml_tensor::TensorShapeError> {
4    let shape = TensorShape::new([2, 3, 4])?;
5
6    assert_eq!(shape.rank(), 3);
7    assert_eq!(shape.num_elements(), Some(24));
8    assert_eq!(TensorDType::Float32.as_str(), "float32");
9    Ok(())
10}