code: def test_repeat_layer():
      inputs = klayers.Input(shape=(2, 3))
      outputs = ilayers.Repeat(8)(inputs)
      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.shape(out) == (2, 8, 2, 3)
      assert np.all(
          out
          == np.array(
              [
                  [
                      [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]],
                      [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]],
                      [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]],
                      [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]],
                      [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]],
                      [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]],
                      [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]],
                      [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]],
                  ],
                  [
                      [[7.0, 8.0, 9.0], [10.0, 11.0, 12.0]],
                      [[7.0, 8.0, 9.0], [10.0, 11.0, 12.0]],
                      [[7.0, 8.0, 9.0], [10.0, 11.0, 12.0]],
                      [[7.0, 8.0, 9.0], [10.0, 11.0, 12.0]],
                      [[7.0, 8.0, 9.0], [10.0, 11.0, 12.0]],
                      [[7.0, 8.0, 9.0], [10.0, 11.0, 12.0]],
                      [[7.0, 8.0, 9.0], [10.0, 11.0, 12.0]],
                      [[7.0, 8.0, 9.0], [10.0, 11.0, 12.0]],
                  ],
              ]
          )
      )
  
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 1 for calling a global function with example usage: test_repeat_layer() without return types