jax.experimental.sparse.bcoo_transpose#

jax.experimental.sparse.bcoo_transpose(mat, *, permutation)[原始碼]#

轉置 BCOO 格式的陣列。

參數:

  • mat (BCOO) – BCOO 格式的陣列。

  • permutation (Sequence[int]) – 一個元組、列表或 ndarray，其中包含 [0,1,..,N-1] 的排列，其中 N 是 mat 的軸數，依序為批次、稀疏和密集維度。 傳回陣列的第 i 個軸對應於 mat 的軸編號 permutation[i]。目前轉置排列不支援將批次軸與非批次軸或密集軸與非密集軸進行排列。

傳回值:

BCOO 格式的陣列。

返回類型:

BCOO