torch.overrides.get_ignored_functions 中文介绍

torch.overrides.get_ignored_functions 是一个用于自定义模块覆盖的函数，它返回一个元组，其中包含应忽略的函数的名称。这些函数将不会被 torch.nn.Module.register_module_forward_pre_hook() 和 torch.nn.Module.register_module_forward_hook() 函数中的自定义覆盖所影响。

在 PyTorch 中，可以通过使用 forward_pre_hook 和 forward_hook 函数来自定义模块的行为。这些函数允许我们在正向传递的不同阶段执行自定义操作，例如记录梯度或修改输入输出。然而，有些函数可能不应该被自定义，例如不确定性操作，这些操作会产生随机结果。在这种情况下，我们可以使用 get_ignored_functions 函数来指定应该忽略的函数列表，以确保它们不会受到自定义操作的影响。

例如，我们可以使用以下代码创建一个自定义模块，其中包含应该忽略的函数列表：

import torch

class MyModule(torch.nn.Module):
    def __init__(self):
        super(MyModule, self).__init__()

    def forward(self, x):
        return x

    @staticmethod
    def get_ignored_functions():
        return ('dropout',)

在这个例子中，我们指定了 dropout 函数应该被忽略。这意味着我们的自定义操作将不会影响 dropout 函数的行为。