jax.numpy.frombuffer

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

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

numpy.frombuffer() 的 JAX 实现。

参数:

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

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

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

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

返回:

一个一维 JAX 数组，表示从缓冲区解释的数据。

返回类型:

Array

另请参阅

• jax.numpy.fromstring(): 将文本字符串转换为一维 JAX 数组。

示例

使用字节缓冲区

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