jax.lax.full_like#

jax.lax.full_like(x, fill_value, dtype=None, shape=None)[source]#

Create a full array like np.full based on the example array x.

Parameters:
Return type:

Array

Returns:

An ndarray with the same shape as x with its entries set equal to fill_value, similar to the output of np.full.