jax.lax.full
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