class_docstr: Tests for linear_operator_psd_kernel.LinearOperatorPSDKernel.
class_docstr_tok: ["Tests","for","linear_operator_psd_kernel.LinearOperatorPSDKernel","."]
code: def test_row_batch(self):
      kernel = exponentiated_quadratic.ExponentiatedQuadratic()
      x1 = tf.random.normal([7, 1, 5, 2])
      x2 = tf.random.normal([1, 3, 4, 2])
      linop = linear_operator_psd_kernel.LinearOperatorPSDKernel(
          kernel, x1, x2
      )
      i = np.random.randint(0, 5, size=(7, 3))
      cov = kernel.matrix(x1, x2)
      expected, actual = self.evaluate(
          [
              tf.gather(cov, i[..., tf.newaxis], batch_dims=2)[
                  ..., 0, :
              ],
              linop.row(i),
          ]
      )
      self.assertAllClose(expected, actual)
  
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 5 for calling the LinearOperatorPSDKernelTest obj's underlying member method code with example usage: obj.test_row_batch() without return types