jax.Array.mT — JAX documentation (original) (raw)

jax.Array.mT#

abstract property Array.mT[source]#

Compute the (batched) matrix transpose.

Refer to jax.numpy.matrix_transpose() for details.