code: def testInnerReshapeSimple(self):
      layer = inner_reshape.InnerReshape([3, 4], [12])
      out = layer(np.arange(2 * 12).reshape(2, 3, 4))
      self.assertAllEqual(
          self.evaluate(out), np.arange(2 * 12).reshape(2, 12)
      )
      out = layer(np.arange(4 * 12).reshape(2, 2, 3, 4))
      self.assertAllEqual(
          self.evaluate(out), np.arange(4 * 12).reshape(2, 2, 12)
      )
  
file_path: agents/tf_agents/keras_layers/inner_reshape_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  import tensorflow
  from tf_agents.keras_layers import inner_reshape
  from tf_agents.utils import test_utils
label_desc: Use image node_id 1 for calling the InnerReshapeTest obj's underlying member method code with example usage: obj.testInnerReshapeSimple() without return types