jax.numpy.append

内容

jax.numpy.append#

jax.numpy.append(arr, values, axis=None)[source]#

返回一个新的数组，其中值追加到原始数组的末尾。

JAX 实现 numpy.append().

参数：

  • arr (ArrayLike) – 原始数组。

  • values (ArrayLike) – 要追加到数组的值。The values 必须与 arr 具有相同的维度数量，并且除了在指定的轴上之外，所有维度都必须匹配。

  • axis (int | None) – 要追加值的轴。如果为 None（默认），则在追加之前 arr 和 values 都将被展平。

返回值：

一个新的数组，其中值追加到 arr。

返回类型：

Array

另请参阅

示例

>>> a = jnp.array([1, 2, 3])
>>> b = jnp.array([4, 5, 6])
>>> jnp.append(a, b)
Array([1, 2, 3, 4, 5, 6], dtype=int32)

沿着特定轴追加

>>> a = jnp.array([[1, 2],
...                [3, 4]])
>>> b = jnp.array([[5, 6]])
>>> jnp.append(a, b, axis=0)
Array([[1, 2],
       [3, 4],
       [5, 6]], dtype=int32)

沿着尾随轴追加

>>> a = jnp.array([[1, 2, 3],
...                [4, 5, 6]])
>>> b = jnp.array([[7], [8]])
>>> jnp.append(a, b, axis=1)
Array([[1, 2, 3, 7],
       [4, 5, 6, 8]], dtype=int32)
目录