code: def testInnerReshapeUnknowns(self):
      layer = inner_reshape.InnerReshape([None, None], [-1])
      out = layer(np.arange(3 * 20).reshape(3, 4, 5))
      self.assertAllEqual(
          self.evaluate(out), np.arange(3 * 20).reshape(3, 20)
      )
      out = layer(np.arange(6 * 20).reshape(2, 3, 4, 5))
      self.assertAllEqual(
          self.evaluate(out), np.arange(6 * 20).reshape(2, 3, 20)
      )
  
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 2 for calling the InnerReshapeTest obj's underlying member method code with example usage: obj.testInnerReshapeUnknowns() without return types