dpctl.tensor.matrix_transpose

dpctl.tensor.matrix_transpose(x)[source]

Transposes the innermost two dimensions of x, where x is a 2-dimensional matrix or a stack of 2-dimensional matrices.

To convert from a 1-dimensional array to a 2-dimensional column vector, use x[:, dpt.newaxis].

Parameters:

x (usm_ndarray) – Input array with shape (…, m, n).

Returns:

Array with shape (…, n, m).

Return type:

usm_ndarray