jax.Array.transpose#

abstractArray.transpose(*args)[source]#

Returns a copy of the array with axes transposed.

Refer tojax.numpy.transpose() for full documentation.

Parameters:

• self (Array)

• args (Any)

Return type:

Array