python 动态hack方法

由于很多时候需要hack forward函数，修改其中的逻辑，目前学习到的有两种修改实例的方法，之后还会有补充。

__get__

class OriginalClass:
    def original_function(self):
        print("这是原始函数的执行内容")

def hijacked_function(self):
    print("这是被劫持替换后的函数执行内容")


if __name__ == "__main__":
    original_obj = OriginalClass()
    # 获取原始函数对象
    original_main_entry = original_obj.original_function
    # 模拟劫持替换
    original_obj.original_function = hijacked_function.__get__(original_obj)
    # 调用被替换后的函数
    original_obj.original_function()
    # 可以验证一下原始函数的绑定情况
    another_obj = OriginalClass()
    another_obj.original_function()

types.MethodType

import types

class MyClass:
    def original_method(self):
        print("执行原始方法")

def new_method(self):
    print("执行替换后的方法")

if __name__ == "__main__":
    obj = MyClass()
    obj.original_method = types.MethodType(new_method, obj)
    obj.original_method()

题外话：pytoch hack

pytorch支持导出、修改Module中间变量

torch.Tensor.register_hook()
torch.nn.Module.register_forward_hook()
torch.nn.Module.register_backward_hook()
torch.nn.Module.register_forward_pre_hook()