Fixed missing axis annotation in tf map_coordinates

Author: Phil2852

keras/src/backend/tensorflow/image.py

@@ -707,7 +707,7 @@ def process_coordinates(coords, size):
         gathered = tf.transpose(tf.gather_nd(input_arr, indices))
 
         if fill_mode == "constant":
-            all_valid = tf.reduce_all(validities)
+            all_valid = tf.reduce_all(validities, axis=0)
             gathered = tf.where(all_valid, gathered, fill_value)
 
         contribution = gathered

keras/src/ops/image_test.py

@@ -1865,6 +1865,28 @@ def test_elastic_transform(self):
         )
         self.assertAllClose(np.var(ref_out), np.var(out), atol=1e-2, rtol=1e-2)
 
+    def test_map_coordinates_constant_padding_tf(self):
+        input_img = tf.ones((2, 2), dtype=tf.uint8)
+        # one pixel outside of the input space around the edges
+        grid = tf.stack(
+            tf.meshgrid(
+                tf.range(-1, 3, dtype=tf.float32),
+                tf.range(-1, 3, dtype=tf.float32),
+                indexing="ij",
+            ),
+            axis=0,
+        )
+        out = kimage.map_coordinates(
+            input_img, grid, order=0, fill_mode="constant", fill_value=0
+        ).numpy()
+
+        # check for ones in the middle and zeros around the edges
+        assert np.all(out[:1] == 0)
+        assert np.all(out[-1:] == 0)
+        assert np.all(out[:, :1] == 0)
+        assert np.all(out[:, -1:] == 0)
+        assert np.all(out[1:3, 1:3] == 1)
+
 
 class ImageOpsBehaviorTests(testing.TestCase):
     def setUp(self):