如何在torchscript中使用自定义python对象

import torch

class MyCustomObject:
    def __init__(self, param1, param2):
        self.param1 = param1
        self.param2 = param2

    def my_method(self, input):
        # Custom logic here
        return output

@torch.jit.script
class TorchScriptCustomObject:
    def __init__(self, param1, param2):
        self.param1 = param1
        self.param2 = param2

    def my_method(self, input):
        # Custom logic here
        return output

import torch

class MyCustomObject:
    def __init__(self, param1, param2):
        self.param1 = param1
        self.param2 = param2

    def my_method(self, input):
        # Custom logic here
        return output

custom_object = MyCustomObject(param1, param2)
torchscript_object = torch.jit.script(custom_object)

import torch

class MyCustomObject:
    def __init__(self, param1, param2):
        self.param1 = param1
        self.param2 = param2

    def my_method(self, input):
        # Custom logic here
        return output

custom_object = MyCustomObject(param1, param2)
torchscript_method = torch.jit.trace(custom_object.my_method, example_input)