besskge.utils.complex_rotation

besskge.utils.complex_rotation(v, r)[source]

Batched rotation by unitary tensors.

Parameters:

  • v (Tensor) – shape: (a, 2*e) Complex tensor to rotate: v[:,:e] real part, v[:,e:] imaginary part.

  • r (Tensor) – shape: (a, e) Rotate v[k] by \(e^{i \pi r[k]}\)

Return type:

Tensor

Returns:

shape: (a, 2*e) Row-wise rotated tensors.