jax.lax.transpose

内容

jax.lax.transpose#

jax.lax.transpose(operand, permutation)[source]#

包装 XLA 的 Transpose 运算符。

参数：:

  • operand (ArrayLike)

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

返回类型：:

Array

内容