def initialize(self, prng_key: jax.Array) -> Parameter[T]:
    """Randomly initializes the parameter.

    Args:
      prng_key: Key to use for initialization.

    Returns:
      A new instance of ``Parameter``.

    Raises:
      ValueError: If the initializer's output does not match the expected
      structure from ``self.value_structure``.
    """
    value = self.initializer(prng_key)
    shapecheck.check_structure(value=value, pattern=self.value_structure)
    return Parameter(self.initializer(prng_key), self.name)
Randomly initializes the parameter.

    Args:
      prng_key: Key to use for initialization.

    Returns:
      A new instance of ``Parameter``.

    Raises:
      ValueError: If the initializer's output does not match the expected
      structure from ``self.value_structure``.
    
penzai/deprecated/v1/nn/parameters.py