code: def testProbWithATolBcastsLoc(self):
      # 3 batch of deterministics on R^2.
      loc = [0.0, 1.0]
      atol = [[0.02], [0.03], [0.05]]
      x = [[0.0, 0.99], [0.04, 1.0], [0.0, 1.045]]
      dist = det.VectorDeterministic(loc, atol=atol, validate_args=True)
      expected_prob = [1.0, 0.0, 1.0]
      prob = dist.prob(x)
      self.assertAllEqual((3,), prob.shape)
      self.assertAllEqual(expected_prob, self.evaluate(prob))
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import deterministic
  from tensorflow_probability.python.distributions import gamma
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.distributions import mvn_diag
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 9 for calling the VectorDeterministicTest obj's underlying member method code with example usage: obj.testProbWithATolBcastsLoc() without return types