//clika-runtime/io.clika.runtime/Ops/bucketize

bucketize

[common]
fun bucketize(input: Tensor, boundaries: Tensor, outInt32: Boolean = false, right: Boolean = false): Tensor

bucketize(input: Tensor, boundaries: Tensor, outInt32: Boolean = false, right: Boolean = false): the bucketize operator. Bucket index of each input element against a sorted 1-D boundaries. With right == false (default) an element in [b[i-1], b[i]) maps to bucket i; right == true uses (b[i-1], b[i]]. Output shape mirrors input.