fouriax.KJonesMatrixLayer#

class KJonesMatrixLayer(jones_matrix, aperture_diameter_um=None)#

Bases: OpticalLayer

k-space Jones matrix modulation layer.

Parameters:

jones_matrix (Array)
aperture_diameter_um (float | None)

__init__(jones_matrix, aperture_diameter_um=None)#

Parameters:

jones_matrix (Array)
aperture_diameter_um (float | None)

Return type:

None

Methods

`__init__`(jones_matrix[, aperture_diameter_um])
`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.

Attributes

`aperture_diameter_um`
`jones_matrix`

jones_matrix: Array#

aperture_diameter_um: float | None = None#

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

validate_for(field)#

Check that field is compatible with this layer.

The default implementation validates the field itself. Subclasses should call super().validate_for(field) and add any layer-specific checks such as domain, polarization, or shape requirements.

Parameters:: field (Field)
Return type:: None

parameters()#

Return trainable or configurable layer parameters.

Returns an empty dict by default. Subclasses override this to expose arrays that should be updated during optimisation.

Return type:: dict[str, Array]