Trait TensorLoader

pub trait TensorLoader<E: Error> {
    // Required methods
    fn load(&mut self, name: &str) -> Result<Tensor, E>;
    fn load_manual(&mut self, name: &str, ne: &[usize]) -> Result<Tensor, E>;
    fn finish(self) -> (Context, HashMap<String, Tensor>, Option<Mmap>);
}

Used by models to fetch tensors from a loader.

Required Methods

fn load(&mut self, name: &str) -> Result<Tensor, E>

Gets a tensor from the loader.

fn load_manual(&mut self, name: &str, ne: &[usize]) -> Result<Tensor, E>

Loads a tensor from the loader.

fn finish(self) -> (Context, HashMap<String, Tensor>, Option<Mmap>)

Finish loading the model, and extract all of the state from the loader.

Implementors