//clika-runtime/io.clika.runtime/Ops/maskedSelect

maskedSelect

[common]
fun maskedSelect(input: Tensor, mask: Tensor): Tensor

maskedSelect(input: Tensor, mask: Tensor): the masked_select operator. The elements of input where mask is true, as a 1-D tensor. The output LENGTH is data-dependent (the number of true entries); under tracing it carries a data-dependent extent. mask is Bool and broadcasts to input's shape.