torch.nn.utils.stateless.functional_call

torch.nn.utils.stateless.functional_call(module, parameters_and_buffers, args=None, kwargs=None, *, tie_weights=True, strict=False)

通过替换模块的参数和缓冲区，在模块上执行功能调用。

警告

该 API 自 PyTorch 2.0 起已弃用，将在 PyTorch 的将来版本中删除。请使用 torch.func.functional_call() 代替，它是该 API 的直接替代品。

注意

如果模块具有活动参数化，则通过将名称设置为常规参数名称的 parameters_and_buffers 参数传递值将完全禁用参数化。如果要将参数化函数应用于传递的值，请将键设置为 {submodule_name}.parametrizations.{parameter_name}.original 。

注意

如果模块在参数/缓冲区上执行就地操作，这些操作将反映在 parameters_and_buffers 输入中。

示例：

>>> a = {'foo': torch.zeros(())}
>>> mod = Foo()  # does self.foo = self.foo + 1
>>> print(mod.foo)  # tensor(0.)
>>> functional_call(mod, a, torch.ones(()))
>>> print(mod.foo)  # tensor(0.)
>>> print(a['foo'])  # tensor(1.)

注意

如果模块具有绑定权重，functional_call 是否尊重绑定由 tie_weights 标志决定。

示例：

>>> a = {'foo': torch.zeros(())}
>>> mod = Foo()  # has both self.foo and self.foo_tied which are tied. Returns x + self.foo + self.foo_tied
>>> print(mod.foo)  # tensor(1.)
>>> mod(torch.zeros(()))  # tensor(2.)
>>> functional_call(mod, a, torch.zeros(()))  # tensor(0.) since it will change self.foo_tied too
>>> functional_call(mod, a, torch.zeros(()), tie_weights=False)  # tensor(1.)--self.foo_tied is not updated
>>> new_a = {'foo': torch.zeros(()), 'foo_tied': torch.zeros(())}
>>> functional_call(mod, new_a, torch.zeros()) # tensor(0.)

参数:

• 模块（torch.nn.Module）- 要调用的模块。

• parameters_and_buffers (str 和 Tensor 的字典) – 模块调用中将使用的参数。

• args (Any 或元组) – 要传递给模块调用的参数。如果不是元组，则视为单个参数。

• kwargs (dict) – 要传递给模块调用的关键字参数

• tie_weights (bool，可选) – 如果为 True，则原始模型中绑定的参数和缓冲区在重新参数化版本中将被视为绑定。因此，如果为 True 且传递给绑定的参数和缓冲区的值不同，将引发错误。如果为 False，则不会尊重原始绑定的参数和缓冲区，除非传递给两个权重的值相同。默认值：True。

• 严格（bool，可选）- 如果为 True，则传入的参数和缓冲区必须与原始模块中的参数和缓冲区匹配。因此，如果为 True 且存在任何缺失或意外的键，则会报错。默认：False。

返回值:

调用 module 的结果。

返回类型:

任何