torch.nn.utils.stateless.functional_call

torch.nn.utils.stateless.functional_call(module, parameters_and_buffers, args, kwargs=None) [source]

Выполняет функциональный вызов модуля, заменяя параметры и буферы модуля предоставленными.

Примечание

Если модуль имеет активные параметризации, передача значения в аргументе 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.)

Параметры:

• module (torch.nn.Module) – вызываемый модуль
• parameters_and_buffers (dict of str and Tensor) – параметры, которые будут использованы при вызове модуля.
• args (tuple) – аргументы, которые будут переданы при вызове модуля
• kwargs (dict) – ключевые аргументы, которые будут переданы при вызове модуля

Возвращаемое значение:

результат вызова module.

Тип возвращаемого значения:

Любой