jax.numpy.empty_like#

jax.numpy.empty_like(prototype,dtype=None,shape=None,*,device=None)[source]#

Create an empty array with the same shape and dtype as an array.

JAX implementation ofnumpy.empty_like(). Because XLA cannot createan un-initialized array,jax.numpy.empty() will always return anarray full of zeros.

Parameters:

• a – Array-like object withshape anddtype attributes.

• shape (Any) – optionally override the shape of the created array.

• dtype (str |type[Any]|dtype |SupportsDType |None) – optionally override the dtype of the created array.

• device (Device |Sharding |None) – (optional)Device orShardingto which the created array will be committed.

• prototype (Array |ndarray |bool |number |bool |int |float |complex |TypedNdArray |DuckTypedArray)

Returns:

Array of the specified shape and dtype, on the specified device if specified.

Return type:

Array

See also

• jax.numpy.empty()

• jax.numpy.zeros_like()

• jax.numpy.ones_like()

• jax.numpy.full_like()

Examples

>>>x=jnp.arange(4)>>>jnp.empty_like(x)Array([0, 0, 0, 0], dtype=int32)>>>jnp.empty_like(x,dtype=bool)Array([False, False, False, False], dtype=bool)>>>jnp.empty_like(x,shape=(2,3))Array([[0, 0, 0],       [0, 0, 0]], dtype=int32)