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 –

  • mode (sharding will currently be ignored in jitted) –

  • change (this might) –

  • future. (in the) –

Return type:

Array