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 按批次（batch）、稀疏（sparse）和稠密（dense）维度顺序计算的轴数。返回数组的第 i 个轴对应于 mat 中编号为 permutation[i] 的轴。当前转置排列不支持将批次轴与非批次轴进行置换，也不支持将稠密轴与非稠密轴进行置换。

返回:

BCOO 格式数组。

返回类型:

BCOO