cr.nimble.mat_transpose

cr.nimble.mat_transpose(x)[source]

Returns the transpose of an array of matrices

Parameters

x (jax.numpy.ndarray) – An nd-array (2 or more dimensions)

Returns

Array with last two dimensions transposed

Return type

(jax.numpy.ndarray)