fouriax.JonesMatrixLayer#

class JonesMatrixLayer(jones_matrix)#

Bases: OpticalLayer

Spatial-domain Jones matrix modulation layer.

Parameters:: jones_matrix (Array)

__init__(jones_matrix)#

Parameters:: jones_matrix (Array)
Return type:: None

Methods

`__init__`(jones_matrix)
`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

jones_matrix

jones_matrix: Array#

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]