jax.lax.transpose

目录

jax.lax.transpose#

jax.lax.transpose(operand, permutation)[源代码][源代码]#

封装了 XLA 的 Transpose 操作符。

参数:

operand (ArrayLike)
permutation (Sequence[int] | np.ndarray)

返回类型: