keras/src/ops/numpy.py/Nonzero/compute_output_spec
class Nonzero:    def compute_output_spec(self, x):
        return tuple(
            [KerasTensor((None,), dtype="int32") for _ in range(len(x.shape))]
        )