Skip to main content

split

onnx_runtime_session::sequence

Function split 

Source 
pub fn split(
    data: &[u8],
    dtype: DataType,
    shape: &[usize],
    axis: i64,
    split: SplitSpec<'_>,
    keepdims: bool,
) -> SequenceResult<SequenceValue>
Expand description

Split raw contiguous tensor bytes into a sequence of session tensors.

keepdims only affects SplitSpec::Each, as required by ONNX; an explicit split input always retains the split axis.