torch.autograd.forward_ad.unpack_dual

torch.autograd.forward_ad.unpack_dual(tensor, *, level=None) [source]

Распаковывает «дуальный тензор», чтобы получить его значение Tensor и градиент обратного распространения AD. Результатом является именованная кортеж (primal, tangent) , где primal является представлением tensor и tangent — тангенс tensor как есть. Ни один из этих тензоров не может быть дуальным тензором уровня level.

Эта функция обратима.

Пример:

>>> with dual_level():
...     inp = make_dual(x, x_t)
...     out = f(inp)
...     y, jvp = unpack_dual(out)
...     jvp = unpack_dual(out).tangent

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