matrix_transpose

matrix_transpose(x: array, /) → array

Transposes a matrix (or a stack of matrices) x.

Parameters:

x (array) – input array having shape (..., M, N) and whose innermost two dimensions form MxN matrices.

Returns:

out (array) – an array containing the transpose for each matrix. The returned array must have shape (..., N, M). The returned array must have the same data type as x.