jax.experimental.sparse.bcoo_transpose

jax.experimental.sparse.bcoo_transpose#

jax.experimental.sparse.bcoo_transpose(mat, *, permutation)[source]#

转置 BCOO 格式的数组。

参数:
  • mat (BCOO) – 一个 BCOO 格式的数组。

  • permutation (Sequence[int]) – 一个元组或列表或 ndarray,其中包含 [0,1,..,N-1] 的排列,其中 N 是 mat 轴的数量,按批次、稀疏和密集维度的顺序排列。返回数组的第 i 个轴对应于 mat 的第 permutation[i] 个轴。转置排列目前不支持将批次轴与非批次轴或将密集轴与非密集轴置换。

返回:

一个 BCOO 格式的数组。

返回类型:

BCOO