code: def expand_tile(value, size):
      """Add a new axis of given size."""
      value = tf.convert_to_tensor(value=value, name="value")
      ndims = value.shape.ndims
      return tf.tile(
          tf.expand_dims(value, axis=0), [size] + [1] * ndims
      )
  
file_path: finetune/finetune/base_models/gpt2/featurizer.py
imports: import functools
  import tensorflow
  from finetune.util.shapes import shape_list, lengths_from_eos_idx
  from finetune.optimizers.recompute_grads import recompute_grad
  from finetune.nn.activations import gelu
  from finetune.base_models.gpt.featurizer import norm, dropout, get_pos_values
label_desc: Use image node_id 10 for calling a global function with example usage: expand_tile(value, size) and returns: tf