Updated random_grayscale.py compute_output_spec (#21326)

* Update nn.py

* Update nn.py

* Update nn.py

* Update nn.py

* Update nn.py

Corrected indentation in doc string

* Update nn.py

* Update random_grayscale.py

Fixed issue with passing a single image without batch dimension.

* Update keras/src/layers/preprocessing/image_preprocessing/random_grayscale.py

Co-authored-by: Jyotinder Singh <[email protected]>

* Update random_grayscale_test.py

Test case for unbatched inputs

* code reformat

* Update random_grayscale_test.py

Testcase for checking both unbatched and batched single image inputs.

* changed compute_output_spec

There was a bug, and it was causing cycle in graph.

* Update random_grayscale.py

removed the use of tree.map_structure

---------

Co-authored-by: Jyotinder Singh <[email protected]>

Author: pctablet505

keras/src/layers/preprocessing/image_preprocessing/random_grayscale.py

@@ -1,5 +1,4 @@
 from keras.src import backend
-from keras.src import tree
 from keras.src.api_export import keras_export
 from keras.src.layers.preprocessing.image_preprocessing.base_image_preprocessing_layer import (  # noqa: E501
     BaseImagePreprocessingLayer,
@@ -97,11 +96,8 @@ def compute_output_shape(self, input_shape):
         return input_shape
 
     def compute_output_spec(self, inputs, **kwargs):
-        return tree.map_structure(
-            lambda x: backend.KerasTensor(
-                x.shape, dtype=x.dtype, sparse=x.sparse
-            ),
-            inputs,
+        return backend.KerasTensor(
+            inputs.shape, dtype=inputs.dtype, sparse=inputs.sparse
         )
 
     def transform_bounding_boxes(self, bounding_boxes, **kwargs):