use-ml-tensor 0.0.1

Tensor shape and metadata primitives for RustUse machine-learning workflows.
Documentation 
1
2
3
4
5
6
7
8
9
10

use use_ml_tensor::{TensorDType, TensorShape};

fn main() -> Result<(), use_ml_tensor::TensorShapeError> {
    let shape = TensorShape::new([2, 3, 4])?;

    assert_eq!(shape.rank(), 3);
    assert_eq!(shape.num_elements(), Some(24));
    assert_eq!(TensorDType::Float32.as_str(), "float32");
    Ok(())
}