code: def test_batch_flatten():
      x = tf.ones([5, 6, 7, 8, 9])
      x_flat = layers_utils.batch_flatten(x)
      assert np.all(tf.shape(x_flat).numpy() == [5 * 6 * 7 * 8, 9])
  
file_path: rasa/tests/utils/tensorflow/test_layers_utils.py
imports: import pytest
  import tensorflow
  import numpy
  from typing import List, Optional, Union
  import rasa.utils.tensorflow.layers_utils
label_desc: Use image node_id 3 for calling a global function with example usage: test_batch_flatten() without return types