jax.numpy.fromstring

jax.numpy.fromstring(string, dtype=<class 'float'>, count=-1, *, sep)

将文本字符串转换为 1-D JAX 数组。

numpy.fromstring() 的 JAX 实现。

参数:

• string (str) – 包含数据的输入字符串。

• dtype (DTypeLike) – 可选。数组的所需数据类型。默认为 float。

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

• sep (str) – 用于分隔输入字符串中值的字符串。

返回:

一个 1 维 JAX 数组，包含从输入字符串解析的数据。

返回类型:

Array

另请参阅

• jax.numpy.frombuffer(): 从实现缓冲区接口的对象构造 JAX 数组。

示例

>>> jnp.fromstring("1 2 3", dtype=int, sep=" ")
Array([1, 2, 3], dtype=int32)
>>> jnp.fromstring("0.1, 0.2, 0.3", dtype=float, count=2, sep=",")
Array([0.1, 0.2], dtype=float32)