jax.numpy.full_like

Contents

jax.numpy.full_like#

jax.numpy.full_like(a, fill_value, dtype=None, shape=None, *, device=None)[source]#

Return a full array with the same shape and type as a given array.

LAX-backend implementation of numpy.full_like().

Original docstring below.

Parameters:
  • a (array_like) – The shape and data-type of a define these same attributes of the returned array.

  • fill_value (array_like) – Fill value.

  • dtype (data-type, optional) – Overrides the data type of the result.

  • shape (int or sequence of ints, optional.) – Overrides the shape of the result. If order=’K’ and the number of dimensions is unchanged, will try to keep order, otherwise, order=’C’ is implied.

  • device (str, optional) – The device on which to place the created array. Default: None. For Array-API interoperability only, so must be "cpu" if passed.

Returns:

out – Array of fill_value with the same shape and type as a.

Return type:

ndarray