docstr: Checks that all vectors form an orthonormal basis.
docstr_tok: ["Checks","that","all","vectors","form","an","orthonormal","basis","."]
code: def testGramSchmidtOrthonormal(self):
      """Checks that all vectors form an orthonormal basis."""
      dim, n_vectors = 10, 5
      vectors = tf.random.normal(
          [dim, n_vectors], seed=test_util.test_seed()
      )
  
      ortho = self.evaluate(gram_schmidt(vectors))
  
      for i in range(n_vectors):
          self.assertAllClose(
              np.linalg.norm(ortho[..., i]), 1.0, rtol=1e-7
          )
          for j in range(i + 1, n_vectors):
              self.assertAllClose(
                  np.dot(ortho[..., i], ortho[..., j]), 0.0, atol=1e-6
              )
  
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 2 for calling the GramSchmidtTest obj's underlying member method code with example usage: obj.testGramSchmidtOrthonormal() without return types