Struct BorrowedTensor 

pub struct BorrowedTensor<'a> { /* private fields */ }

Implementations

impl<'a> BorrowedTensor<'a>

pub fn from_f32(_data: &'a [f32], shape: &[usize]) -> Result<Self>

Examples found in repository

examples/load_and_predict.rs (line 18)

fn main() {
    let model_path = std::env::args()
        .nth(1)
        .unwrap_or_else(|| "tests/fixtures/test_linear.mlmodelc".to_string());

    println!("Loading model: {model_path}");
    let model = Model::load(&model_path, ComputeUnits::All).expect("Failed to load model");

    // Create a simple input tensor
    let input_data = vec![1.0f32, 2.0, 3.0, 4.0];
    let tensor =
        BorrowedTensor::from_f32(&input_data, &[1, 4]).expect("Failed to create tensor");

    println!("Running prediction...");
    let prediction = model.predict(&[("input", &tensor)]).expect("Prediction failed");

    let (output, shape) = prediction.get_f32("output").expect("Failed to get output");
    println!("Output shape: {shape:?}");
    println!("Output data: {output:?}");
}

pub fn from_i32(_data: &'a [i32], shape: &[usize]) -> Result<Self>

pub fn from_f64(_data: &'a [f64], shape: &[usize]) -> Result<Self>

pub fn from_f16_bits(_data: &'a [u16], shape: &[usize]) -> Result<Self>

pub fn from_i16(_data: &'a [i16], shape: &[usize]) -> Result<Self>

pub fn from_i8(_data: &'a [i8], shape: &[usize]) -> Result<Self>

pub fn from_u32(_data: &'a [u32], shape: &[usize]) -> Result<Self>

pub fn from_u16(_data: &'a [u16], shape: &[usize]) -> Result<Self>

pub fn from_u8(_data: &'a [u8], shape: &[usize]) -> Result<Self>

pub fn shape(&self) -> &[usize]

pub fn data_type(&self) -> DataType

pub fn element_count(&self) -> usize

pub fn size_bytes(&self) -> usize

Trait Implementations

impl<'a> Debug for BorrowedTensor<'a>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl AsMultiArray for BorrowedTensor<'_>

Available on non-Apple only.