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.