full

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

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.