jax.numpy.append

jax.numpy.append(arr, values, axis=None)

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

JAX 实现的 numpy.append()。

参数:

• arr (ArrayLike) – 原始数组。

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

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

返回值:

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

返回类型:

数组

另请参阅

• jax.numpy.insert()

• jax.numpy.delete()

示例

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