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 (Any) – the value to fill the new array with.
dtype (Optional[Any]) – the type of the output array, or None. If not None, fill_value
will be cast to dtype.
- Return type
Any
