code: def test_variable_shaped_tensor_array_uniform_dim():
      shape1 = (3, 2, 2)
      shape2 = (3, 4, 4)
      a1 = np.arange(np.prod(shape1)).reshape(shape1)
      a2 = np.arange(np.prod(shape2)).reshape(shape2)
      ta = TensorArray([a1, a2])
      assert len(ta) == 2
      assert ta.is_variable_shaped
      for a, expected in zip(ta.to_numpy(), [a1, a2]):
          np.testing.assert_array_equal(a, expected)
  
imports: import itertools
  import numpy
  import pandas
  from packaging.version import parse
  import pyarrow
  import pytest
  from ray.air.util.tensor_extensions.arrow import ArrowTensorArray, ArrowTensorType, ArrowVariableShapedTensorArray, ArrowVariableShapedTensorType
  from ray.air.util.tensor_extensions.pandas import TensorArray, TensorDtype
  from ray.air.util.tensor_extensions.utils import create_ragged_ndarray
  from ray._private.utils import _get_pyarrow_version
label_desc: Use image node_id 26 for calling a global function with example usage: test_variable_shaped_tensor_array_uniform_dim() without return types