code: def initializer_record(
      name, initializer, dtypes, min_dims=2, max_dims=4
  ):
      shapes = [
          shape
          for shape in ALL_SHAPES
          if min_dims <= len(shape) <= max_dims
      ]
      return InitializerRecord(name, initializer, shapes, dtypes)
  
imports: import collections
  from functools import partial
  import itertools
  import unittest
  from absl.testing import absltest
  from absl.testing import parameterized
  import scipy.stats
  from jax._src import config
  from jax._src import core
  from jax._src import test_util
  from jax._src import ad_checkpoint
  from jax.test_util import check_grads
  from jax import nn
  from jax import random
  import jax
  import jax.numpy
label_desc: Use image node_id 1 for calling a global function with example usage: initializer_record(name, initializer, dtypes, min_dims, max_dims) and returns: InitializerRecord