jax.numpy.concatenate

jax.numpy.concatenate(arrays, axis=0, dtype=None)

沿现有轴连接一系列数组。

LAX-backend 实现的 numpy.concatenate()。

原始文档字符串如下。

参数:

• axis (int, optional) – 数组将被连接的轴。如果轴为 None，数组在使用前会被展平。默认值为 0。

• dtype (str or dtype) – 如果提供，目标数组将具有此数据类型。不能与 out 一起提供。

• arrays (np.ndarray | Array | Sequence[ArrayLike])

返回:

res – 连接后的数组。

返回类型:

ndarray