tf.keras.ops.nonzero

Return the indices of the elements that are non-zero.

x Input tensor.

Indices of elements that are non-zero.