torch.autograd.forward_ad.unpack_dual

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

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

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

Пример:

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

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