How can I create a custom class in c++ and import the output in python? Riight now, the example given here https://pytorch.org/tutorials/advanced/torch_script_custom_ops.htm
