jax.numpy.empty_like

内容

jax.numpy.empty_like#

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

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

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

参数:

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

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

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

  • 设备 (xc.Device | 分片 | 无 | 无) – (可选) Device 或 Sharding，创建的数组将提交到该设备。

  • 原型 (类数组 | DuckTypedArray)

返回值:

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

返回类型:

数组

另请参见

示例

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