code: def _compute_and_check_ntks(f, x1, x2, params):
      ntks = _get_ntks(f, x1, x2, params, vmap_axes=None)
      ntks_vmap = _get_ntks(f, x1, x2, params, vmap_axes=0)
      _check_ntks(ntks + ntks_vmap)
  
imports: from absl import app
  import neural_tangents
  import tensorflow
label_desc: Use image node_id 3 for calling a global function with example usage: _compute_and_check_ntks(f, x1, x2, params) without return types