class_docstr: Tests for linear_operator_psd_kernel.LinearOperatorPSDKernel.
class_docstr_tok: ["Tests","for","linear_operator_psd_kernel.LinearOperatorPSDKernel","."]
code: def test_shape(self):
      kernel = exponentiated_quadratic.ExponentiatedQuadratic(
          amplitude=tf.random.uniform([17, 1, 1]), feature_ndims=2
      )
      x1 = tf.random.normal([1, 11, 5, 2, 13])
      x2 = tf.random.normal([7, 1, 3, 2, 13])
      linop = linear_operator_psd_kernel.LinearOperatorPSDKernel(
          kernel, x1, x2
      )
      self.assertAllEqual((17, 7, 11, 5, 3), linop.shape)
      self.assertAllEqual((17, 7, 11), linop.batch_shape)
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.linalg import linear_operator_psd_kernel
  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 polynomial
label_desc: Use image node_id 1 for calling the LinearOperatorPSDKernelTest obj's underlying member method code with example usage: obj.test_shape() without return types