jax.numpy.empty_like#
- jax.numpy.empty_like(prototype, dtype=None, shape=None, *, device=None)[source]#
Return a new array with the same shape and type as a given array.
LAX-backend implementation of
numpy.empty_like().Because XLA cannot create uninitialized arrays, the JAX version will return an array initialized with zeros.
Original docstring below.
- Parameters:
prototype (array_like) – The shape and data-type of prototype define these same attributes of the returned array.
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 (xc.Device | Sharding | None)
- Returns:
out – Array of uninitialized (arbitrary) data with the same shape and type as prototype.
- Return type:
ndarray