pdt_model = TestNNModule()
     # Runs the pdt_model in eager model with the inputs provided and annotates the arguments of forward
     scripted_model = torch.jit.script(pdt_model, example_inputs={pdt_model: [([10, 20, ], ), ], })

     # Run the scripted_model with actual inputs
     print(scripted_model([20]))
