Skip to main content

InferenceSession

beat_this::runtime

Trait InferenceSession 

Source 
pub trait InferenceSession {
    // Required method
    fn run(
        &mut self,
        inputs: &[(&str, &Tensor)],
    ) -> Result<HashMap<String, Tensor>>;
}
Expand description

A loaded model session ready for inference.

Required Methods§

Source

fn run(&mut self, inputs: &[(&str, &Tensor)]) -> Result<HashMap<String, Tensor>>

Run inference with named inputs, return named outputs.

Implementors§