dual_level

class torch.autograd.forward_ad.dual_level [source]

Контекст-менеджер, который включает прямой AD. Все вычисления прямого AD должны выполняться в dual_level контексте.

Примечание

Контекст dual_level должным образом входит и выходит из двойного уровня, чтобы управлять текущим уровнем прямого AD, который используется по умолчанию другими функциями в этом API.

В настоящее время мы не планируем поддерживать вложенные dual_level контексты, поэтому поддерживается только один уровень прямого AD. Для вычисления градиентов высших порядков можно использовать functorch’s jvp.

Пример:

>>> x = torch.tensor([1])
>>> x_t = torch.tensor([1])
>>> with dual_level():
...   inp = make_dual(x, x_t)
...   # Do computations with inp
...   out = your_fn(inp)
...   _, grad = unpack_dual(out)
>>> grad is None
False
>>> # After exiting the level, the grad is deleted
>>> _, grad_after = unpack_dual(out)
>>> grad is None
True

Подробные инструкции по использованию этого API см. в учебном пособии по прямому AD.