Skip to main content

//clika-runtime/io.clika.runtime/Ops/splitBySize

splitBySize

[common]
fun splitBySize(input: Tensor, chunkSize: Long, dim: Long = 0): List<Tensor>

splitBySize(input: Tensor, chunkSize: Long, dim: Long = 0L): the split_by_size operator. Split along dim into pieces of chunk_size, ceil(extent / chunk_size) of them, the last possibly shorter. Returns VIEWS sharing the source's storage (no copy).