set_grad_enabled

class torch.autograd.grad_mode.set_grad_enabled(mode) [исходный код]

Менеджер контекста, который включает или отключает вычисление градиентов.

set_grad_enabled включает или отключает вычисление градиентов в зависимости от аргумента mode. Его можно использовать как менеджер контекста или как функцию.

Этот менеджер контекста действует только в пределах текущего потока; он не влияет на вычисления в других потоках.

Параметры:

mode (bool) – Флаг, определяющий, следует ли включить вычисление градиентов (True) или отключить его (False). Это можно использовать для условного включения градиентов.

Примечание

set_grad_enabled — один из нескольких механизмов, позволяющих локально включать или отключать вычисление градиентов. Дополнительные сведения об их сравнении см. в разделе Локальное отключение вычисления градиентов.

Примечание

Этот API не распространяется на прямой режим AD.

Пример::

>>> x = torch.tensor([1.], requires_grad=True)
>>> is_train = False
>>> with torch.set_grad_enabled(is_train):
...     y = x * 2
>>> y.requires_grad
False
>>> _ = torch.set_grad_enabled(True)
>>> y = x * 2
>>> y.requires_grad
True
>>> _ = torch.set_grad_enabled(False)
>>> y = x * 2
>>> y.requires_grad
False

clone() [исходный код]

Создает копию этого класса

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

set_grad_enabled