Contents:
Returns the transpose of an array of matrices
x (jax.numpy.ndarray) – An nd-array (2 or more dimensions)
Array with last two dimensions transposed
(jax.numpy.ndarray)