code: def test_reducemean_layer():
      inputs = klayers.Input(shape=(2, 3))
      repeated = ilayers.Repeat(8)(inputs)
      outputs = ilayers.ReduceMean()(repeated)
      model = kmodels.Model(
          inputs=inputs, outputs=outputs, name="TestRepeat"
      )
  
      x = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])
      out = model.predict(x)
  
      assert np.all(out == x)
  
file_path: innvestigate/tests/backend/test_layer.py
imports: from __future__ import annotations
  import numpy
  import pytest
  import tensorflow.keras.layers
  import tensorflow.keras.models
  import innvestigate.layers
  from innvestigate.analyzer.gradient_based import Gradient
label_desc: Use image node_id 2 for calling a global function with example usage: test_reducemean_layer() without return types