jax.experimental.sparse.bcoo_转置

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