code: def testBatchResultSize(self):
      # 3x3x3 matrices, 3x3x1 right-hand sides.
      matrix = np.array(
          [1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0] * 3
      ).reshape(
          3, 3, 3
      )  # pylint: disable=too-many-function-args
      rhs = np.array([1.0, 2.0, 3.0] * 3).reshape(
          3, 3, 1
      )  # pylint: disable=too-many-function-args
      answer = linalg_ops.matrix_solve(matrix, rhs)
      ls_answer = linalg_ops.matrix_solve_ls(matrix, rhs)
      self.assertEqual(ls_answer.get_shape(), [3, 3, 1])
      self.assertEqual(answer.get_shape(), [3, 3, 1])
  
imports: import numpy
  from tensorflow.python.client import session
  from tensorflow.python.eager import context
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import control_flow_ops
  from tensorflow.python.ops import linalg_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import benchmark
  from tensorflow.python.platform import test
label_desc: Use image node_id 4 for calling the MatrixSolveLsOpTest obj's underlying member method code with example usage: obj.testBatchResultSize() without return types