full

Contents

full#

penzai.core.named_axes.full(named_shape: Mapping[AxisName, int], fill_value: jax.typing.ArrayLike, dtype: np.DTypeLike | None = None) → NamedArray[source]#

Constructs a full named array with a given shape.

Parameters:

  • named_shape – Named shape for the result.

  • fill_value – Value to fill the array with.

  • dtype – Optional dtype for the result. If not provided, will be inferred from fill_value.

Returns:

NamedArray with the given named shape, filled with fill_value.

Contents