code: def get_constant_linear_params(num_outputs):
      w = np.ones([_INPUT_DIM, num_outputs])
      b = np.zeros([num_outputs])
      return dict(w=w, b=b)
  
imports: import functools
  from absl.testing import absltest
  from absl.testing import parameterized
  import chex
  import haiku
  from haiku import data_structures
  from haiku import initializers
  import jax
  import jax.nn
  import jax.numpy
  import numpy
  from rlax._src import pop_art
label_desc: Use image node_id 2 for calling a global function with example usage: get_constant_linear_params(num_outputs) and returns: dict