code: def test_expand_dims_dynamic(self):
      x = random_ops.random_uniform([3, 5])
      axis = 1
      num_axes = constant_op.constant([2])
      expected = array_ops.reshape(x, [3, 1, 1, 5])
      actual = pfor._expand_dims(x, axis, num_axes)
      self.assertAllEqual(expected, actual)
  
file_path: tensorflow-upstream/tensorflow/python/ops/parallel_for/pfor_test.py
imports: from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops.parallel_for import pfor
  from tensorflow.python.platform import test
label_desc: Use image node_id 6 for calling the PForTest obj's underlying member method code with example usage: obj.test_expand_dims_dynamic() without return types