jax.lax.fullΒΆ

jax.lax.full(shape, fill_value, dtype=None)[source]ΒΆ

Returns an array of shape filled with fill_value.

Parameters
  • shape (Sequence[int]) – 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