class_docstr: Tests the correctness of the Y = inv(tril) transformation.
class_docstr_tok: ["Tests","the","correctness","of","the","Y","=","inv","(","tril",")","transformation","."]
code: def testBatch(self):
      # Test batch computation with input shape (2, 1, 2, 2), i.e. batch shape
      # (2, 1).
      inv = matrix_inverse_tril.MatrixInverseTriL(validate_args=True)
      x_ = np.array(
          [[[[1.0, 0.0], [2.0, 3.0]]], [[[4.0, 0.0], [5.0, -6.0]]]],
          dtype=np.float32,
      )
      x_inv_ = self._inv(x_)
  
      y = inv.forward(x_)
      x_back = inv.inverse(x_inv_)
  
      y_, x_back_ = self.evaluate([y, x_back])
  
      self.assertAllClose(x_inv_, y_, atol=0.0, rtol=1e-5)
      self.assertAllClose(x_, x_back_, atol=0.0, rtol=1e-5)
  
file_path: probability/tensorflow_probability/python/bijectors/matrix_inverse_tril_test.py
imports: import numpy
  from tensorflow_probability.python.bijectors import bijector_test_util
  from tensorflow_probability.python.bijectors import fill_triangular
  from tensorflow_probability.python.bijectors import invert
  from tensorflow_probability.python.bijectors import matrix_inverse_tril
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 5 for calling the MatrixInverseTriLBijectorTest obj's underlying member method code with example usage: obj.testBatch() without return types