code: def _slice_layers_params(layers_params):
      sliced_layers_params = {}
      for k, v in layers_params.items():
          for inner_k in v:
              for var_slice, suffix in zip(v[inner_k], suffixes):
                  k_new = k.split("/")[-1] + suffix
                  if k_new not in sliced_layers_params:
                      sliced_layers_params[k_new] = {}
                  sliced_layers_params[k_new][inner_k] = var_slice
      return sliced_layers_params
  
imports: import functools
  import re
  from typing import Optional
  from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import base
  from haiku._src import basic
  from haiku._src import config
  from haiku._src import initializers
  from haiku._src import layer_stack
  from haiku._src import module
  from haiku._src import multi_transform
  from haiku._src import transform
  from haiku._src import utils
  import jax
  import jax.numpy
  import numpy
  from scipy import stats
label_desc: Use image node_id 1 for calling a global function with example usage: _slice_layers_params(layers_params) and returns: sliced_layers_params