diff --git a/keras/src/ops/numpy.py b/keras/src/ops/numpy.py index 4588c446770..c11899720ae 100644 --- a/keras/src/ops/numpy.py +++ b/keras/src/ops/numpy.py @@ -3881,7 +3881,7 @@ def call(self, x): return backend.numpy.nonzero(x) def compute_output_spec(self, x): - return KerasTensor([None] * len(x.shape)) + return KerasTensor([None] * len(x.shape), dtype="int32") @keras_export(["keras.ops.nonzero", "keras.ops.numpy.nonzero"]) diff --git a/keras/src/ops/numpy_test.py b/keras/src/ops/numpy_test.py index 3960c6f0226..4e8e18062cd 100644 --- a/keras/src/ops/numpy_test.py +++ b/keras/src/ops/numpy_test.py @@ -7102,7 +7102,10 @@ def test_nonzero(self, dtype): self.assertEqual( standardize_dtype(knp.nonzero(x)[0].dtype), expected_dtype ) - # TODO: verify Nonzero + self.assertEqual( + standardize_dtype(knp.Nonzero().symbolic_call(x)[0].dtype), + expected_dtype, + ) @parameterized.named_parameters( named_product(dtypes=itertools.combinations(ALL_DTYPES, 2))