class_docstr: Tests for methods in linear_interpolation module.
docstr: Tests linear interpolation with multiple batching dimensions.
docstr_tok: ["Tests","linear","interpolation","with","multiple","batching","dimensions","."]
code: def test_linear_interpolation_multiple_batching_dimensions(
      self, optimize_for_tpu
  ):
      """Tests linear interpolation with multiple batching dimensions."""
      for dtype in (np.float32, np.float64):
          x = np.array([[[1.5], [3.5]]], dtype=dtype)
          x_data = np.array([[[1, 2], [3, 4]]], dtype=dtype)
          y_data = np.array([[[0, 1], [2, 3]]], dtype=dtype)
          result = self.evaluate(
              tff.math.interpolation.linear.interpolate(
                  x, x_data, y_data, optimize_for_tpu=optimize_for_tpu
              )
          )
          self.assertEqual(result.dtype, dtype)
          self.assertAllClose(result, np.array([[[0.5], [2.5]]]), 1e-8)
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  import tf_quant_finance
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 8 for calling the LinearInterpolation obj's underlying member method code with example usage: obj.test_linear_interpolation_multiple_batching_dimensions(optimize_for_tpu) without return types