See the compiled graph as Python code
     print(scripted_fn.code)

     # Call the function using the TorchScript interpreter
     scripted_fn(20, 100)

Scripting an nn.Module
      Scripting an "nn.Module" by default will compile the "forward"
      method and recursively compile any methods, submodules, and
      functions called by "forward". If a "nn.Module" only uses
      features supported in TorchScript, no changes to the original
      module code should be necessary. "script" will construct
      "ScriptModule" that has copies of the attributes, parameters,
      and methods of the original module.
  Example (scripting a simple module with a Parameter):

     import torch

     class MyModule(torch.nn.Module):
         def __init__(self, N, M):
             super(MyModule, self).__init__()
             # This parameter will be copied to the new ScriptModule
             self.weight = torch.nn.Parameter(torch.rand(N, M))
