dq.trace

trace(x: QArrayLike) -> Array

Returns the trace of a qarray along its last two dimensions.

Parameters:

• x (qarray-like of shape (..., n, n)) –

  Qarray-like.

Returns:

• (array of shape (...)) –

  Trace of x.

Examples:

>>> x = jnp.ones((3, 3))
>>> dq.trace(x)
Array(3., dtype=float32)