partial_trace

qujax.partial_trace(densitytensor, indices_to_trace)

Traces out (discards) specified qubits, resulting in a densitytensor representing the mixed quantum state on the remaining qubits.

Parameters:

• densitytensor (ndarray) – Input densitytensor.

• indices_to_trace (Sequence[int]) – Indices of qubits to trace out/discard.

Returns:

Resulting densitytensor on remaining qubits.

Return type:

ndarray