jax.numpy.append#

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

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

numpy.append() 的 JAX 实现。

参数:

  • arr (ArrayLike) – 原始数组。

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

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

返回:

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

返回类型:

数组

另请参阅

示例

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