class_docstr: Tests correctness of the Y = scale @ x transformation.
code: def testBatch(self, is_static):
      # Corresponds to 1 2x2 matrix, with twos on the diagonal.
      scale_diag = [[2.0, 2]]
      bijector = scale_matvec_diag.ScaleMatvecDiag(
          scale_diag=scale_diag
      )
      x = self.maybe_static([[[1.0, 1]]], is_static)
      self.assertAllClose([[[2.0, 2]]], bijector.forward(x))
      self.assertAllClose([[[0.5, 0.5]]], bijector.inverse(x))
      self.assertAllClose(
          [-np.log(4)],
          bijector.inverse_log_det_jacobian(x, event_ndims=1),
      )
  
file_path: probability/tensorflow_probability/python/bijectors/scale_matvec_diag_test.py
imports: from absl.testing import parameterized
  import numpy
  from tensorflow_probability.python.bijectors import scale_matvec_diag
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 2 for calling the ScaleMatvecDiagTest obj's underlying member method code with example usage: obj.testBatch(is_static) without return types