1import torch 2from torch import nn 3 4 5class NeuralNetwork(nn.Module): 6 def forward(self, x): 7 return torch.add(x, 10) 8 9 10model = NeuralNetwork() 11script = torch.jit.script(model) 12torch.jit.save(script, "aot_test_model.pt") 13