code: def test_lambda_basics(self):
      self.run_layer_test(
          layers.Lambda,
          init_kwargs={
              "function": ops.square,
          },
          input_shape=(2, 3),
          expected_output_shape=(2, 3),
          expected_num_trainable_weights=0,
          expected_num_non_trainable_weights=0,
          expected_num_seed_generators=0,
          expected_num_losses=0,
          supports_masking=False,
          custom_objects={"square": ops.square},
      )
      self.run_layer_test(
          layers.Lambda,
          init_kwargs={
              "function": ops.square,
              "mask": ops.ones((2, 3)),
          },
          input_shape=(2, 3, 4),
          expected_output_shape=(2, 3, 4),
          expected_num_trainable_weights=0,
          expected_num_non_trainable_weights=0,
          expected_num_seed_generators=0,
          expected_num_losses=0,
          supports_masking=True,
          custom_objects={"square": ops.square},
      )
  
      def stacker(x):
          return ops.concatenate([x, x], axis=1)
  
      self.run_layer_test(
          layers.Lambda,
          init_kwargs={"function": stacker, "output_shape": (6,)},
          input_shape=(2, 3),
          expected_output_shape=(2, 6),
          expected_num_trainable_weights=0,
          expected_num_non_trainable_weights=0,
          expected_num_seed_generators=0,
          expected_num_losses=0,
          supports_masking=False,
          custom_objects={"stacker": stacker},
      )
  
      def stacker_shape(s):
          return (s[0], s[1] * 2)
  
      self.run_layer_test(
          layers.Lambda,
          init_kwargs={
              "function": stacker,
              "output_shape": stacker_shape,
          },
          input_shape=(2, 3),
          expected_output_shape=(2, 6),
          expected_num_trainable_weights=0,
          expected_num_non_trainable_weights=0,
          expected_num_seed_generators=0,
          expected_num_losses=0,
          supports_masking=False,
          custom_objects={
              "stacker": stacker,
              "stacker_shape": stacker_shape,
          },
      )
  
file_path: keras/keras/layers/core/lambda_layer_test.py
imports: import numpy
  import pytest
  from keras import layers
  from keras import ops
  from keras import testing
label_desc: Use image node_id 1 for calling the LambdaTest obj's underlying member method code with example usage: obj.test_lambda_basics() and returns: ops, s,  unknown