docstr: Tests the Jacobian of the Chamfer distance loss.
docstr_tok: ["Tests","the","Jacobian","of","the","Chamfer","distance","loss","."]
code: def test_chamfer_distance_evaluate_jacobian(self):
      """Tests the Jacobian of the Chamfer distance loss."""
      point_set_a, point_set_b = _random_point_sets()
  
      with self.subTest(name="jacobian_wrt_point_set_a"):
          self.assert_jacobian_is_correct_fn(
              lambda x: chamfer_distance.evaluate(x, point_set_b),
              [point_set_a],
              atol=1e-5,
          )
  
      with self.subTest(name="jacobian_wrt_point_set_b"):
          self.assert_jacobian_is_correct_fn(
              lambda x: chamfer_distance.evaluate(point_set_a, x),
              [point_set_b],
              atol=1e-5,
          )
  
file_path: graphics/tensorflow_graphics/nn/loss/tests/chamfer_distance_test.py
imports: from absl.testing import parameterized
  import numpy
  from tensorflow_graphics.nn.loss import chamfer_distance
  from tensorflow_graphics.util import test_case
label_desc: Use image node_id 2 for calling the ChamferDistanceTest obj's underlying member method code with example usage: obj.test_chamfer_distance_evaluate_jacobian() without return types