jax.experimental.sparse.bcoo_concatenate

目錄

jax.experimental.sparse.bcoo_concatenate#

jax.experimental.sparse.bcoo_concatenate(operands, *, dimension)[原始碼]#

jax.lax.concatenate() 的稀疏實作

參數:

operands (Sequence[BCOO]) – 要串聯的 BCOO 陣列序列。陣列必須具有相等的形狀，除了 dimension 軸之外。此外，陣列必須具有等效的批次、稀疏和密集維度。
dimension (int) – 指定要沿其串聯陣列的維度的正整數。維度必須在輸入的批次或稀疏維度中；不支援沿密集維度串聯。

返回:

包含輸入串聯的 BCOO 陣列。

返回類型: