jax.lax.concatenate

jax.lax.concatenate(operands, dimension)

沿dimension 连接一系列数组。

包装了XLA的Concatenate运算符。

参数:

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

• dimension ( int ) – 沿着该维度连接数组。

返回:

包含连接的数组。

返回类型:

Array