code: def testOutputShape(self):
      for axis in [1, -1]:
          with self.cached_session():
              tensor = array_ops.placeholder(
                  dtypes.float32, shape=[None, 12]
              )
              size_splits = [3, 7, 2]
              outputs = array_ops.split(tensor, size_splits, axis)
              for i, output in enumerate(outputs):
                  self.assertEqual(
                      output.get_shape().as_list(),
                      [None, size_splits[i]],
                  )
  
imports: import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gradients_impl
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 12 for calling the SplitOpTest obj's underlying member method code with example usage: obj.testOutputShape() without return types