torch.jit.trace_module
torch.jit.trace_module(mod, inputs, optimize=None, check_trace=True, check_inputs=None, check_tolerance=1e-05, strict=True, _force_outplace=False, _module_class=None, _compilation_unit=, example_inputs_is_kwarg=False)
Trace a module and return an executable "ScriptModule" that will be
   optimized using just-in-time compilation. When a module is passed
   to "torch.jit.trace", only the "forward" method is run and traced.
   With "trace_module", you can specify a dictionary of method names
   to example inputs to trace (see the "inputs") argument below.
See "torch.jit.trace" for more information on tracing.
Parameters:
      * mod (torch.nn.Module) -- A "torch.nn.Module" containing
        methods whose names are specified in "inputs". The given
        methods will be compiled as a part of a single ScriptModule.
  * **inputs** (*dict*) -- A dict containing sample inputs indexed
