code: def _numpy_exp_quad_matrix(amplitude, length_scale, x, feature_ndims):
      return _numpy_exp_quad(
          amplitude,
          length_scale,
          np.expand_dims(x, -feature_ndims - 2),
          np.expand_dims(x, -feature_ndims - 1),
          feature_ndims,
      )
  
file_path: probability/tensorflow_probability/python/math/psd_kernels/feature_transformed_test.py
imports: from absl.testing import parameterized
  import numpy
  from tensorflow_probability.python.bijectors import scale
  from tensorflow_probability.python.bijectors import scale_matvec_diag
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math.psd_kernels import exponentiated_quadratic
  from tensorflow_probability.python.math.psd_kernels import feature_transformed
  from tensorflow_probability.python.math.psd_kernels.internal import util
label_desc: Use image node_id 2 for calling a global function with example usage: _numpy_exp_quad_matrix(amplitude, length_scale, x, feature_ndims) and returns: _numpy_exp_quad