//clika-runtime/io.clika.runtime/Ops/nonzero

nonzero

[common]
fun nonzero(input: Tensor): Tensor

nonzero(input: Tensor): the nonzero operator. Coordinates of the nonzero elements: an [n, ndim] Int64 matrix, one row per nonzero element of input. An element is nonzero exactly when its Bool cast is true: +0 and -0 are zero; a subnormal, an infinity and a NaN are nonzero. The row count is data-dependent (see masked_select); a 1-arg where() call is sugar for this op.