torch.autograd.forward_ad.unpack_dual

torch.autograd.forward_ad.unpack_dual(tensor, *, level=None) [исходный код]

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

Результат представляет собой именованный кортеж (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

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