Skip to main content

//clika-runtime/io.clika.runtime/Ops/splitWithSizes

splitWithSizes

[common]
fun splitWithSizes(input: Tensor, sizes: LongArray, dim: Long = 0): List<Tensor>

splitWithSizes(input: Tensor, sizes: LongArray, dim: Long = 0L): the split_with_sizes operator. Split along dim into chunks of the given lengths; sizes must sum to the dim's extent. Each length is a literal or a 0-D integer Tensor. Returns VIEWS; every chunk shares the source's storage (no copy).