code: def _RemoveAxes12():
      """Returns a layer that removes two internal size-1 axes from an array."""
      return tl.Fn("RemoveAxes12", lambda x: jnp.squeeze(x, (1, 2)))
  
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 3 for calling a global function with example usage: _RemoveAxes12() and returns: tl