code: def test_docstring_example(self):
      add = lambda a, b: a + b
      add_vector_to_scalar = vectorization_util.make_rank_polymorphic(
          add, core_ndims=(1, 0)
      )
      self.assertAllEqual(
          [[4.0, 5.0], [5.0, 6.0], [6.0, 7.0]],
          self.evaluate(
              add_vector_to_scalar(
                  tf.constant([1.0, 2.0]), tf.constant([3.0, 4.0, 5.0])
              )
          ),
      )
  
imports: import warnings
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import lognormal
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.distributions import poisson
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.internal import vectorization_util
label_desc: Use image node_id 3 for calling the VectorizationTest obj's underlying member method code with example usage: obj.test_docstring_example() without return types