jax.numpy.empty#
- jax.numpy.empty(shape, dtype=None, *, device=None)[source]#
Create an empty array.
JAX implementation of
numpy.empty()
. Because XLA cannot create an un-initialized array,jax.numpy.empty()
will always return an array full of zeros.- Parameters:
shape (Any) – int or sequence of ints specifying the shape of the created array.
dtype (str | type[Any] | dtype | SupportsDType | None) – optional dtype for the created array; defaults to floating point.
device (Device | Sharding | None) – (optional)
Device
orSharding
to which the created array will be committed.
- Returns:
Array of the specified shape and dtype, on the specified device if specified.
- Return type:
Examples
>>> jnp.empty(4) Array([0., 0., 0., 0.], dtype=float32) >>> jnp.empty((2, 3), dtype=bool) Array([[False, False, False], [False, False, False]], dtype=bool)