jax.lax.concatenate

jax.lax.concatenate(operands, dimension)

沿 dimension 连接一系列数组。

包装 XLA 的 Concatenate 运算符。

参数:

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

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

返回:

包含连接的数组。

返回类型:

Array