torchsig.transforms.functional.nonlinear_amplifier

torchsig.transforms.functional.nonlinear_amplifier(data: ndarray, gain: float = 1.0, psat_backoff: float = 10.0, phi_rad: float = 0.0, auto_scale: bool = True) → ndarray[source]

A memoryless AM/AM, AM/PM nonlinear amplifier function-based model using a hyperbolic tangent output power response defined by gain and saturation power.

Parameters:

  • data (np.ndarray) – Complex valued IQ data samples.

  • gain (float) – Small-signal linear gain. Default 1.0.

  • psat_backoff (float) – Saturated output power factor relative to the input signal mean power. For example, operating at a 2.0 psat_backoff factor with a 1 W mean power signal has saturation power level at 2.0 W. Default 10.0.

  • phi_rad (float) – Signal relative phase shift at saturation (radians). Modeled to vary linearly from (0.0 rad, 0.0 power). Default 0.0 rad.

  • auto_scale (bool) – Automatically rescale output power to match full-scale peak input power prior to transform, based on peak estimates. Default True.

Returns:

Nonlinearly distorted IQ data.

Return type:

np.ndarray