Skip to main content

batchify

tract_core::batchify

Function batchify 

Source 
pub fn batchify(
    model: &TypedModel,
    batch: &Symbol,
    shared: &[String],
) -> TractResult<TypedModel>
Expand description

Batch axis on axis 0 of every input but the shared ones, sized by batch.