code: def test_shape_inference_stack_concat(self):
      arr = tensor_array_ops.TensorArray(size=4, dtype=dtypes.float32)
      new_arr = arr.write(0, np.ones((2, 3)))
      self.assertEqual(new_arr.stack().shape, (4, 2, 3))
      self.assertEqual(new_arr.concat().shape, (8, 3))
  
file_path: tensorflow-upstream/tensorflow/python/ops/tensor_array_ops_test.py
imports: import numpy
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import tensor_array_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 5 for calling the TensorArrayOpsTest obj's underlying member method code with example usage: obj.test_shape_inference_stack_concat() without return types