jax.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