Update torch_script_custom_ops.rst (#1767)

Duplicate of #888. Fixes #910.
pytorch · Dec 15, 2021 · aecb73d · aecb73d
1 parent 657f138
commit aecb73d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/advanced_source/torch_script_custom_ops.rst b/advanced_source/torch_script_custom_ops.rst
@@ -579,13 +579,13 @@ custom operator, that loads and executes a serialized TorchScript model:
     }
 
     // Deserialize the ScriptModule from a file using torch::jit::load().
-    std::shared_ptr<torch::jit::script::Module> module = torch::jit::load(argv[1]);
+    torch::jit::script::Module module = torch::jit::load(argv[1]);
 
     std::vector<torch::jit::IValue> inputs;
     inputs.push_back(torch::randn({4, 8}));
     inputs.push_back(torch::randn({8, 5}));
 
-    torch::Tensor output = module->forward(std::move(inputs)).toTensor();
+    torch::Tensor output = module.forward(std::move(inputs)).toTensor();
 
     std::cout << output << std::endl;
   }