jax.lax.concatenate

jax.lax.concatenate(operands, dimension)

沿 dimension 连接一系列数组。

包装 XLA 的 Concatenate 运算符。

参数:

• operands (Array | Sequence[ArrayLike]) – 要连接的数组序列。除了在 dimension 轴上，这些数组必须具有相同的形状。

• dimension (int) – 沿其连接数组的维度。

返回:

一个包含连接结果的数组。

返回类型:

Array