prepare_qat_fx
class torch.quantization.quantize_fx.prepare_qat_fx(model, qconfig_mapping, example_inputs, prepare_custom_config=None, backend_config=None)
Prepare a model for quantization aware training
Parameters:
      * model (***) -- torch.nn.Module model
  * **qconfig_mapping** (***) -- see "prepare_fx()"

  * **example_inputs** (***) -- see "prepare_fx()"

  * **prepare_custom_config** (***) -- see "prepare_fx()"

  * **backend_config** (***) -- see "prepare_fx()"

Returns:
      A GraphModule with fake quant modules (configured by
      qconfig_mapping and backend_config), ready for quantization
      aware training
Return type:
      ObservedGraphModule
Example:
  import torch
  from torch.ao.quantization import get_default_qat_qconfig_mapping
  from torch.ao.quantization import prepare_fx

  class Submodule(torch.nn.Module):
      def __init__(self):
          super().__init__()
