code: def split_states(x, n):
      """Reshape the last dimension of x into [n, x.shape[-1]/n]."""
      *start, m = shape_list(x)
      return tf.reshape(x, start + [n, m // n])
  
imports: import numpy
  import tensorflow
label_desc: Use image node_id 6 for calling a global function with example usage: split_states(x, n) and returns: tf