Skip to content

dq.unit

unit(x: ArrayLike) -> Array

Normalize a ket, bra or density matrix to unit norm.

The returned object is divided by its norm (see dq.norm()).

Parameters

  • x (array_like of shape (..., n, 1) or (..., 1, n) or (..., n, n))

    Ket, bra or density matrix.

Returns

(array of shape (..., n, 1) or (..., 1, n) or (..., n, n)) Normalized ket, bra or density matrix.

Examples

>>> psi = dq.fock(4, 0) + dq.fock(4, 1)
>>> dq.norm(psi)
Array(1.414, dtype=float32)
>>> psi = dq.unit(psi)
>>> dq.norm(psi)
Array(1., dtype=float32)