code: def testBatched(self):
      shp = (5, 13, 7)
      mat_numpy = self.evaluate(
          tf.random.normal(shp, seed=test_util.test_seed())
      )
      mat = tf.constant(mat_numpy)
      self.assertAllClose(
          np.vectorize(gs_numpy, signature="(d,n)->(d,n)")(mat_numpy),
          gram_schmidt(mat),
      )
  
file_path: probability/tensorflow_probability/python/math/gram_schmidt_test.py
imports: import numpy
  import scipy.linalg
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math.gram_schmidt import gram_schmidt
label_desc: Use image node_id 5 for calling the GramSchmidtTest obj's underlying member method code with example usage: obj.testBatched() without return types