Function split_with_sizes

pub fn split_with_sizes<B: Backend, K: TensorKind<B> + BasicOps<B>>(
    tensor: K::Primitive,
    split_sizes: Vec<usize>,
    dim: usize,
) -> Vec<K::Primitive>

Splits the tensor along the given dimension into chunks with sizes in dim according to split_sizes.

Arguments

• tensor - The tensor.
• split_sizes - Vector of sizes for each chunk.
• dim - The dimension along which to split the tensor.

Returns

A vector of tensors.

Remarks

Fallback solution for backends with no equivalent functionality.