code: def test_docstring_example_passing_fn_arg(self):
      def apply_binop(fn, a, b):
          return fn(a, b)
  
      apply_binop_to_vector_and_scalar = (
          vectorization_util.make_rank_polymorphic(
              apply_binop, core_ndims=(None, 1, 0)
          )
      )
      r = self.evaluate(
          apply_binop_to_vector_and_scalar(
              lambda a, b: a * b,
              tf.constant([1.0, 2.0]),
              tf.constant([3.0, 4.0, 5.0]),
          )
      )
      self.assertAllEqual(
          r,
          np.array(
              [[3.0, 6.0], [4.0, 8.0], [5.0, 10.0]], dtype=np.float32
          ),
      )
  
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 11 for calling the VectorizationTest obj's underlying member method code with example usage: obj.test_docstring_example_passing_fn_arg() and returns: fn