Current behaviour is to drop the non-linear map and output the transformed affine map. This extension is to be able to retain the non-linear component and transform just the linear component.