sun_diffusion.gens module

Hard-coded generators for the SU(2) Lie algebra.

sun_diffusion.gens.pauli(i)

Retrieves the i^th Pauli matrix as a PyTorch tensor.

Note

The pauli matrices are normalized by a factor of \(\frac{1}{\sqrt{2}}\) so that they form an orthonormal basis for \({\rm SU}(2)\), i.e., \({\rm tr}[\sigma^a \sigma^b] = \delta^{ab}\).

Parameters:

i (int) – Index of pauli matrix to get

Return type:

Tensor

Returns:

Pauli matrix i as a PyTorch tensor