complex

Module

Implements the Complex type.

You can import these APIs from the complex package. For example:

from complex import ComplexSIMD

Aliases:

• ComplexFloat32 = ComplexSIMD[f32, 1]

• ComplexFloat64 = ComplexSIMD[f64, 1]

ComplexSIMD

Represents a complex SIMD value.

The class provides basic methods for manipulating complex values.

Parameters:

• type (DType): DType of the value.
• size (Int): SIMD width of the value.

Fields:

• re (SIMD[type, size]): The real part of the complex SIMD value.

• im (SIMD[type, size]): The imaginary part of the complex SIMD value.

Functions:

__init__

__init__(re: SIMD[type, size], im: SIMD[type, size]) -> Self

__neg__

__neg__(self: Self) -> Self

Negates the complex value.

Returns:

The negative of the complex value.

__add__

__add__(self: Self, rhs: Self) -> Self

Adds two complex values.

Args:

• rhs (Self): Complex value to add.

Returns:

A sum of this and RHS complex values.

__mul__

__mul__(self: Self, rhs: Self) -> Self

Multiplies two complex values.

Args:

• rhs (Self): Complex value to multiply with.

Returns:

A product of this and RHS complex values.

norm

norm(self: Self) -> SIMD[type, size]

Returns the magnitude of the complex value.

Returns:

Value of sqrt(re*re + im*im).

squared_norm

squared_norm(self: Self) -> SIMD[type, size]

Returns the squared magnitude of the complex value.

Returns:

Value of re*re + im*im.

fma

fma(self: Self, b: Self, c: Self) -> Self

Computes FMA operation.

Compute fused multiple-add with two other complex values: result = self * b + c

Args:

• b (Self): Multiplier complex value.
• c (Self): Complex value to add.

Returns:

Computed Self * B + C complex value.

squared_add

squared_add(self: Self, c: Self) -> Self

Computes Square-Add operation.

Compute Self * Self + C.

Args:

• c (Self): Complex value to add.

Returns:

Computed Self * Self + C complex value.