tf.keras.ops.nonzero

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

Main aliases

tf.keras.ops.numpy.nonzero

x Input tensor.

Indices of elements that are non-zero.