jax.numpy.frombuffer

内容

jax.numpy.frombuffer#

jax.numpy.frombuffer(buffer, dtype=<class 'float'>, count=-1, offset=0)[source]#

将缓冲区转换为一维 JAX 数组。

JAX 实现 numpy.frombuffer()。

参数：

  • buffer (bytes | Any) – 包含数据的对象。它必须是长度为数据类型元素大小整数倍的 bytes 对象，或者必须是导出 Python 缓冲区接口 的对象。

  • dtype (DTypeLike) – 可选。数组所需的数据类型。默认为 float64。这指定用于解析缓冲区的数据类型，但请注意，解析后，如果 jax_enable_x64 标志设置为 False，则 64 位值将转换为 32 位 JAX 数组。

  • count (int) – 可选整数，指定从缓冲区读取的项目数。如果为 -1（默认值），则读取缓冲区中的所有项目。

  • offset (int) – 可选整数，指定在缓冲区开头跳过的字节数。默认为 0。

返回值：

表示缓冲区中解释数据的 1 维 JAX 数组。

返回类型：

Array

参见

示例

使用字节缓冲区

>>> buf = b"\x00\x01\x02\x03\x04"
>>> jnp.frombuffer(buf, dtype=jnp.uint8)
Array([0, 1, 2, 3, 4], dtype=uint8)
>>> jnp.frombuffer(buf, dtype=jnp.uint8, offset=1)
Array([1, 2, 3, 4], dtype=uint8)

通过 Python 缓冲区接口构建 JAX 数组，使用 Python 内置的 array 模块。

>>> from array import array
>>> pybuffer = array('i', [0, 1, 2, 3, 4])
>>> jnp.frombuffer(pybuffer, dtype=jnp.int32)
Array([0, 1, 2, 3, 4], dtype=int32)
内容