Skip to main content

infer_batch_graph

Function infer_batch_graph 

Source
pub fn infer_batch_graph(
    graph: &mut Graph,
    model: &SequentialModel,
    input: Tensor,
) -> Result<Tensor, ModelError>
Expand description

Runs inference through the autograd graph and returns the output tensor value.