jax.Array.transpose — JAX documentation (original) (raw)

Contents

• Array.transpose()

jax.Array.transpose#

abstract Array.transpose(*args)[source]#

Returns a copy of the array with axes transposed.

Refer to jax.numpy.transpose() for full documentation.

Parameters:

• self (Array)
• args (Any)

Return type:

Array