jax.lax.concatenate

jax.lax.concatenate(operands, dimension)

沿着指定维度连接一系列数组。

包装 XLA 的 Concatenate 运算符。

参数：

• operands (Array | Sequence[ArrayLike]) – 要连接的数组序列。除了指定维度之外，这些数组必须具有相同的形状。

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

返回：

包含连接结果的数组。

返回类型：

Array