jax.lax.concatenate

目录

jax.lax.concatenate#

jax.lax.concatenate(operands, dimension)[源代码][源代码]#

沿 dimension 维度连接一系列数组。

封装了 XLA 的 Concatenate 操作符。

参数:

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

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

返回:

包含连接结果的数组。

返回类型:

Array

目录