jax.numpy.empty

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

创建一个空数组。

numpy.empty() 的 JAX 实现。由于 XLA 无法创建未初始化的数组，jax.numpy.empty() 将始终返回一个充满零的数组。

参数:

• shape (Any) – 指定所创建数组形状的整数或整数序列。

• dtype (DTypeLike | None | None) – 可选的所创建数组的数据类型；默认为浮点型。

• device (xc.Device | Sharding | None | None) – （可选）Device 或 Sharding，用于指定所创建数组将被提交到的设备。

返回:

返回指定形状和数据类型的数组，如果指定了设备，则在指定的设备上。

返回类型:

数组 (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)