code: def _XYAvg():
      """Returns a layer that computes the element-wise average of two arrays."""
      return tl.Fn("XYAvg", lambda x, y: (x + y) / 2.0)
  
imports: import functools
  from trax import layers
  from trax.fastmath import numpy
  from trax.models.reformer import reformer
  from trax.models.research import configurable_transformer
  from trax.models.research import transformer2
label_desc: Use image node_id 5 for calling a global function with example usage: _XYAvg() and returns: tl