jax.experimental.sparse.bcoo_concatenate

jax.experimental.sparse.bcoo_concatenate(operands, *, dimension)

稀疏实现 jax.lax.concatenate()

参数:

• operands (Sequence[BCOO]) – 要连接的 BCOO 数组序列。数组的形状必须相同，但 dimension 轴除外。此外，数组必须具有等效的批次、稀疏和稠密维度。

• dimension (int) – 指定要连接数组的维度正整数。该维度必须是输入的批次或稀疏维度之一；不支持沿着稠密维度进行连接。

返回:

包含输入连接的 BCOO 数组。

返回类型:

BCOO