code: def testSolveBatch(self):
      matrix = np.array([[1.0, 2.0], [3.0, 4.0]])
      rhs = np.array([[1.0, 0.0, 1.0], [0.0, 1.0, 1.0]])
      # Batch of 2x3x2x2 matrices, 2x3x2x3 right-hand sides.
      self._verifySolveAllWaysReal(matrix, rhs, batch_dims=[2, 3])
      # Batch of 3x2x2x2 matrices, 3x2x2x3 right-hand sides.
      self._verifySolveAllWaysReal(matrix, rhs, batch_dims=[3, 2])
  
      matrix = np.array(
          [
              [1.0, 2.0, 3.0, 4.0],
              [-1.0, -2.0, -3.0, -4.0],
              [-1.0, 1.0, 2.0, 3.0],
          ]
      )
      rhs = np.array([[-1.0, 2.0], [1.0, 1.0], [0.0, 1.0], [2.0, 3.0]])
      # Batch of 2x3x4x4 matrices with 3 bands, 2x3x4x2 right-hand sides.
      self._verifySolveAllWaysReal(matrix, rhs, batch_dims=[2, 3])
      # Batch of 3x2x4x4 matrices with 3 bands, 3x2x4x2 right-hand sides.
      self._verifySolveAllWaysReal(matrix, rhs, batch_dims=[3, 2])
  
file_path: tensorflow-upstream/tensorflow/python/kernel_tests/math_ops/banded_triangular_solve_op_test.py
imports: import numpy
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import linalg_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 8 for calling the BandedTriangularSolveOpTest obj's underlying member method code with example usage: obj.testSolveBatch() without return types