jax.lax.transpose#

jax.lax.transpose(operand, permutation)[原始碼]#

包裝 XLA 的 Transpose 運算子。

參數:

  • operand (ArrayLike)

  • permutation (Sequence[int] | np.ndarray)

回傳類型:

Array