code: def get_rel_att_inputs(d_model, n_heads):
      """Global relative attentions bias initialization shared across layers."""
      assert d_model % n_heads == 0 and d_model % 2 == 0
      d_head = d_model // n_heads
  
      bias_initializer = init.RandomNormalInitializer(1e-6)
      context_bias_layer = core.Weights(
          bias_initializer, shape=(1, n_heads, 1, d_head)
      )
      location_bias_layer = core.Weights(
          bias_initializer, shape=(1, n_heads, 1, d_head)
      )
      return context_bias_layer, location_bias_layer
  
imports: from trax import fastmath
  from trax.fastmath import numpy
  from trax.layers import base
  from trax.layers import combinators
  from trax.layers import core
  from trax.layers import initializers
  from trax.layers.assert_shape import assert_shape
  from trax.layers.attention import MergeHeads
  from trax.layers.attention import SplitIntoHeads
label_desc: Use image node_id 2 for calling a global function with example usage: get_rel_att_inputs(d_model, n_heads) and returns: context_bias_layer,  location_bias_layer