code: def testFlatten(self):
      batch_size = 10
      inputs = tf.ones([batch_size, 2, 3, 4, 5])
      mod = reshape.Reshape(output_shape=[-1])
      output = mod(inputs)
      self.assertEqual(output.shape, [batch_size, 2 * 3 * 4 * 5])
  
imports: from absl.testing import parameterized
  import numpy
  from sonnet.src import reshape
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 8 for calling the ReshapeTest obj's underlying member method code with example usage: obj.testFlatten() without return types