jax.experimental.sparse.bcoo_连接# jax.experimental.sparse.bcoo_concatenate(operands, *, dimension)[源代码][源代码]# 稀疏实现 jax.lax.concatenate() 参数: operands (Sequence[BCOO]) – 要连接的 BCOO 数组的序列。数组必须在 dimension 轴之外具有相同的形状。此外，数组必须具有等效的批次、稀疏和密集维度。 dimension (int) – 正整数，指定沿哪个维度连接数组。该维度必须是输入的批次或稀疏维度之一；不支持沿密集维度进行连接。 返回: 包含输入连接的 BCOO 数组。 返回类型: BCOO
