tf.experimental.numpy.nonzero

TensorFlow variant of NumPy's nonzero.

See the NumPy documentation for numpy.nonzero.