Skip to main content

predict_batch

Function predict_batch 

Source
pub fn predict_batch(
    model: &TlModel,
    input: &TlTensor,
    batch_size: usize,
) -> Result<TlTensor, String>
Expand description

Batch prediction: split input into batches, predict, reassemble.