jax.numpy.empty_like

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

创建一个与数组具有相同形状和 dtype 的空数组。

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

参数:

• a – 具有 shape 和 dtype 属性的类数组对象。

• shape (Any | None) – 可选地覆盖创建的数组的形状。

• dtype (DTypeLike | None | None) – 可选地覆盖创建的数组的 dtype。

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

• prototype (ArrayLike | DuckTypedArray)

返回:

在指定设备（如果已指定）上，具有指定形状和数据类型的数组。

返回类型:

数组

另请参阅

• jax.numpy.empty()

• jax.numpy.zeros_like()

• jax.numpy.ones_like()

• jax.numpy.full_like()

示例

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