jax.lax.full#

jax.lax.full(shape, fill_value, dtype=None)[source]#

Returns an array of shape filled with fill_value.

Parameters

shape (Sequence[Union[int, Any]]) – sequence of integers, describing the shape of the output array.
fill_value (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – the value to fill the new array with.
dtype (Union[Any, str, dtype, SupportsDType, None]) – the type of the output array, or None. If not None, fill_value will be cast to dtype.

Return type

Array