Ctrl+K

jax.lax.full

Contents

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) – the type of the output array, or None. If not None, fill_value will be cast to dtype.

  • sharding (Sharding | 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:

Array

Contents