jax.experimental.sparse.bcoo_concatenate

jax.experimental.sparse.bcoo_concatenate#

jax.experimental.sparse.bcoo_concatenate(operands, *, dimension)[source]#

Sparse 实现 of jax.lax.concatenate()

参数:

  • operands (Sequence[BCOO]) – 要连接的 BCOO 数组序列。这些数组必须具有相同的形状，除了在 dimension 轴上。此外，这些数组必须具有等效的批处理、稀疏和密集维度。

  • dimension (int) – 指定要连接数组的维度的正整数。该维度必须位于输入的批处理或稀疏维度中；不支持沿着密集维度连接。

返回值:

包含输入连接结果的 BCOO 数组。

返回值类型:

BCOO

目录