besskge.utils.complex_multiplication

besskge.utils.complex_multiplication(v1, v2)[source]

Batched complex multiplication.

Parameters:

  • v1 (Tensor) – shape: (a, 2*e) v1[:,:e] real part, v1[:,e:] imaginary part.

  • v2 (Tensor) – shape: (a, 2*e) v2[:,:e] real part, v2[:,e:] imaginary part.

Return type:

Tensor

Returns:

shape: (a, 2*e) Row-wise complex multiplication.