jax.numpy.empty

jax.numpy.empty(shape, dtype=None, *, device=None)

建立一個空的陣列。

JAX 版本的 numpy.empty()。由於 XLA 無法建立未初始化的陣列，jax.numpy.empty() 總是會回傳一個充滿零的陣列。

參數:

• shape (Any) – 指定建立陣列形狀的整數或整數序列。

• dtype (DTypeLike | None | None) – 建立陣列的可選 dtype；預設為浮點數。

• device (xc.Device | Sharding | None | None) – (可選) 將建立的陣列提交到的 Device 或 Sharding。

回傳:

指定形狀和 dtype 的陣列，如果指定了裝置，則在指定的裝置上。

回傳型別:

Array

參見

• jax.numpy.empty_like()

• jax.numpy.zeros()

• jax.numpy.ones()

• jax.numpy.full()

範例

>>> 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)