class_docstr: Test the Polynomial kernel.
class_docstr_tok: ["Test","the","Polynomial","kernel","."]
code: def testShapesAreCorrectMatrix(
      self, feature_ndims, x_shape, y_shape, matrix_shape
  ):
      k = polynomial.Polynomial(
          bias_amplitude=0.0,
          slope_amplitude=1.0,
          shift=0.0,
          exponent=1.0,
          feature_ndims=feature_ndims,
      )
      x = np.ones(x_shape, np.float32)
      y = np.ones(y_shape, np.float32)
      self.assertAllEqual(matrix_shape, k.matrix(x, y).shape)
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
  from tensorflow_probability.python.math.psd_kernels import polynomial
label_desc: Use image node_id 12 for calling the PolynomialTest obj's underlying member method code with example usage: obj.testShapesAreCorrectMatrix(feature_ndims, x_shape, y_shape, matrix_shape) without return types