Added docstring ops/image/resize

keras_core/ops/image.py

@@ -55,7 +55,35 @@ def compute_output_spec(self, image):
 def resize(
     image, size, method="bilinear", antialias=False, data_format="channels_last"
 ):
-    # TODO: add docstring
+    """Resize images to size using the specified method.
+
+    Args:
+        image: Input image or batch of images. Must be 3D or 4D.
+        size: Size of output image in (height, width) format.
+        method: Interpolation method. Defaults to 'bilinear'.
+        antialias: Whether to use an antialiasing filter when downsampling an
+            image.
+        data_format: Format of the image tensor. Defaults to 'channels_last'.
+
+    Returns:
+        Resized image or batch of images.
+
+    Examples:
+
+    >>> x = np.random.random((2, 4, 4, 3)) # batch of 2 RGB images
+    >>> y = keras_core.ops.image.resize(x, (2, 2))
+    >>> y.shape
+
+    >>> x = np.random.random((4, 4, 3)) # single RGB image
+    >>> y = keras_core.ops.image.resize(x, (2, 2))
+    >>> y.shape
+
+    >>> x = np.random.random((2, 3, 4, 4)) # batch of 2 RGB images
+    >>> y = keras_core.ops.image.resize(x, (2, 2),
+    ...     data_format='channels_first')
+    >>> y.shape
+    """
+
     if any_symbolic_tensors((image,)):
         return Resize(
             size, method=method, antialias=antialias, data_format=data_format