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[str,type[Any],dtype,SupportsDType,None]) – the type of the output array, or None. If not None, fill_value will be cast to dtype.
- Return type: