code: def testVectorizedMap(self):
      def fn(x):
          y = constant_op.constant([3.0, 4.0])
          # Make a [2, N, N] shaped operator.
          x = x * y[..., array_ops.newaxis, array_ops.newaxis]
          operator = linalg.LinearOperatorFullMatrix(x, is_square=True)
          return operator
  
      x = np.random.uniform(-1.0, 1.0, size=[3, 5, 5]).astype(
          np.float32
      )
      batched_operator = control_flow_ops.vectorized_map(
          fn, ops.convert_to_tensor(x)
      )
      self.assertIsInstance(batched_operator, linalg.LinearOperator)
      self.assertAllEqual(batched_operator.batch_shape, [3, 2])
  
imports: import numpy
  from tensorflow.python.eager import context
  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_shape
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import linalg_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops.linalg import linalg
  from tensorflow.python.ops.parallel_for import control_flow_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 21 for calling the LinearOperatorTest obj's underlying member method code with example usage: obj.testVectorizedMap() and returns: operator