code: def NaiveUpsampling(
      shorten_factor, d_model, *args, **kwargs
  ):  # pylint: disable = unused-argument
      return core.Fn(
          "Repeat", lambda x: jnp.repeat(x, shorten_factor, axis=1)
      )
  
file_path: trax/trax/layers/research/resampling.py
imports: from trax.fastmath import numpy
  from trax.layers import combinators
  from trax.layers import core
  from trax.layers.normalization import LayerNorm
  from trax.layers.pooling import AvgPool
  from trax.layers.research.rel_attention import RelativeAttentionLMLayer
label_desc: Use image node_id 4 for calling a global function with example usage: NaiveUpsampling(shorten_factor, d_model) and returns: core