class_docstr: Base of VarLenTensorGeneratorBase that generates tf.sparse.SparseTensor.
code: def _get_dim(self):
      if self._dim_dist == DimensionDistribution.ONE_DIM:
          return 1
      elif self._dim_dist == DimensionDistribution.TWO_DIM:
          return 2
      elif self._dim_dist == DimensionDistribution.LARGE_DIM:
          return np.random.randint(
              *DIM_DISTRIBUTION_TO_RANGE[self._dim_dist]
          )
      else:
          raise ValueError(
              f"Found unsupported dimension distribution {self._dim_dist}"
          )
  
imports: import enum
  import hashlib
  import numpy
  import tensorflow
  from tests.test_atds_avro.utils.generator.generator_base import Generator
  from tests.test_atds_avro.utils.hash_util import int_to_bytes
label_desc: Use image node_id 2 for calling the VarLenTensorGeneratorBase obj's underlying member method code with example usage: obj._get_dim() and returns: int, int, np