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 的新数组。

返回类型:

Array

另请参阅

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