jax.numpy.frombuffer#

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

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

JAX 对 numpy.frombuffer() 的实现。

参数:

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

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

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

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

返回:

一个一维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)