jax.numpy.fromstring#

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

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

JAX 实现的 numpy.fromstring().

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

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

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

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

返回:

一个包含从输入字符串解析的数据的一维 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)