jax.numpy.fromstring

内容

jax.numpy.fromstring#

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

将文本字符串转换为一维 JAX 数组。

JAX 实现 numpy.fromstring().

参数：

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

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

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

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

返回值：

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

返回类型：

Array

另请参阅

示例

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