code: def build_tensors(dim):
      """Setup basic test with a flat volatility surface."""
      year = dim * [[2021, 2022]]
      month = dim * [[1, 1]]
      day = dim * [[1, 1]]
      expiries = tff.datetime.dates_from_year_month_day(
          year, month, day
      )
      valuation_date = [(2020, 1, 1)]
      expiry_times = tff.datetime.daycount_actual_365_fixed(
          start_date=valuation_date, end_date=expiries, dtype=tf.float64
      )
      strikes = dim * [
          [[0.8, 0.9, 1.0, 1.1, 1.3], [0.8, 0.9, 1.0, 1.1, 1.5]]
      ]
      strikes = tf.constant(strikes, dtype=tf.float64)
      iv = dim * [
          [[0.1, 0.1, 0.1, 0.1, 0.1], [0.1, 0.1, 0.1, 0.1, 0.1]]
      ]
      spot = dim * [1.0]
      spot = tf.constant(spot, dtype=tf.float64)
      return valuation_date, expiries, expiry_times, strikes, iv, spot
  
imports: import functools
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  import tf_quant_finance
  from tf_quant_finance.experimental.local_volatility import local_volatility_model
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 1 for calling a global function with example usage: build_tensors(dim) and returns: valuation_date,  expiries,  expiry_times,  strikes,  iv,  spot