code: def get_test_data_extended():
      logits = np.array(
          [
              [
                  [0, 0, 0.5, 0.5, 0.2],
                  [0, 0, 0.3, 0.3, 0.1],
                  [0, 0, 0.9, 10, 1],
              ],
              [
                  [0, 0, 0.2, 0.5, 0.2],
                  [0, 0, 3, 0.3, 0.1],
                  [0, 0, 0.9, 1, 1],
              ],
          ]
      )
      tags = np.array([[2, 3, 4], [3, 2, 2]])
  
      transitions = np.array(
          [
              [0.1, 0.2, 0.3, 0.4, 0.5],
              [0.8, 0.3, 0.1, 0.7, 0.9],
              [-0.3, 2.1, -5.6, 3.4, 4.0],
              [0.2, 0.4, 0.6, -0.3, -0.4],
              [1.0, 1.0, 1.0, 1.0, 1.0],
          ]
      )
  
      boundary_values = np.ones((5,))
      crf_layer = CRF(
          units=5,
          use_kernel=False,  # disable kernel transform
          chain_initializer=tf.keras.initializers.Constant(transitions),
          use_boundary=True,
          boundary_initializer=tf.keras.initializers.Constant(
              boundary_values
          ),
          name="crf_layer",
      )
      return logits, tags, transitions, boundary_values, crf_layer
  
imports: import itertools
  import os
  import math
  import tempfile
  import pytest
  import numpy
  import tensorflow
  from tensorflow_addons.layers.crf import CRF
  from tensorflow_addons.text.crf import crf_log_likelihood
  from tensorflow_addons.utils import test_utils
label_desc: Use image node_id 2 for calling a global function with example usage: get_test_data_extended() and returns: logits,  tags,  transitions,  boundary_values,  crf_layer