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.
pub fn infer_batch_graph(
graph: &mut Graph,
model: &SequentialModel,
input: Tensor,
) -> Result<Tensor, ModelError>Runs inference through the autograd graph and returns the output tensor value.