jax.lax.concatenate

jax.lax.concatenate(operands, dimension)

沿著 dimension 連接陣列序列。

包裝 XLA 的 Concatenate 運算子。

參數:

• operands (Array | Sequence[ArrayLike]) – 要連接的陣列序列。陣列必須具有相等的形狀，除了 dimension 軸之外。

• dimension (int) – 沿著此維度連接陣列。

返回:

包含連接的陣列。

返回類型:

Array