fouriax.FourierTransform#

class FourierTransform#

Bases: OpticalLayer

Explicit spatial → k-space domain transform layer.

__init__()#
Return type:

None

Methods

__init__()

forward(field)

Apply this layer's optical transformation to field.

parameters()

Return trainable or configurable layer parameters.

validate_for(field)

Check that field is compatible with this layer.

forward(field)#

Apply this layer’s optical transformation to field.

Parameters:

field (Field) – Input optical field to transform.

Returns:

Transformed field. Subclasses should preserve batch axes, wavelength axes, and metadata unless the layer explicitly changes them.

Return type:

Field