Trait TensorDataAccess 

pub trait TensorDataAccess {
    // Required methods
    fn data_f32(&self) -> Option<&[f32]>;
    fn data_bytes(&self) -> Option<&[u8]>;
    fn to_vec_f32(&self) -> Result<Vec<f32>>;
    fn to_vec_u8(&self) -> Result<Vec<u8>>;
}

Tensor data access for interop

Required Methods

fn data_f32(&self) -> Option<&[f32]>

Get read-only access to raw data (CPU only) Returns None if tensor is not on CPU or data is not contiguous

fn data_bytes(&self) -> Option<&[u8]>

Get read-only access to raw data as bytes

fn to_vec_f32(&self) -> Result<Vec<f32>>

Copy tensor data to a Vec (may involve device-to-host transfer)

fn to_vec_u8(&self) -> Result<Vec<u8>>

Copy tensor data to a Vec

Implementors