torch.autograd.forward_ad.make_dual

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

Связывает значение тензора с прямым градиентом, касательной, для создания «дуального тензора», который используется для вычисления градиентов прямой AD. Результатом является новый тензор, связанный с tensor с tangent встроенным как атрибут как есть, если у него такая же структура хранения, или скопированным в противном случае. Атрибут касательной можно восстановить с помощью unpack_dual().

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

Учитывая функцию f, чья якобианная матрица равна J, это позволяет вычислить произведение якобиана на вектор (jvp) между J и заданным вектором v следующим образом.

Пример:

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

Обратитесь к руководству по forward-mode AD для подробных шагов по использованию этого API.