jax.lax.full#
- jax.lax.full(shape, fill_value, dtype=None, *, sharding=None)[source]#
Returns an array of shape filled with fill_value.
- Parameters:
shape (Shape) – sequence of integers, describing the shape of the output array.
fill_value (ArrayLike) – the value to fill the new array with.
dtype (DTypeLike | None | None) – the type of the output array, or None. If not None, fill_value will be cast to dtype.
sharding (Sharding | None | None) – an optional sharding specification for the resulting array, note, sharding will currently be ignored in jitted mode, this might change in the future.
- Return type: