code: def testBatch(self, is_static):
      bijector = shift.Shift([[2.0, -0.5], [1.0, -3.0]])
      x = self.maybe_static([1.0, 1.0], is_static)
  
      self.assertAllClose(
          [[3.0, 0.5], [2.0, -2.0]], bijector.forward(x)
      )
      self.assertAllClose(
          [[-1.0, 1.5], [0.0, 4.0]], bijector.inverse(x)
      )
      self.assertAllClose(
          0.0, bijector.inverse_log_det_jacobian(x, event_ndims=1)
      )
  
imports: from absl.testing import parameterized
  from tensorflow_probability.python.bijectors import shift
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 2 for calling the ShiftTest obj's underlying member method code with example usage: obj.testBatch(is_static) without return types