torch.func.grad_and_value

torch.func.grad_and_value(func, argnums=0, has_aux=False) [исходный код]

Возвращает функцию для вычисления кортежа, содержащего градиент и первичное (прямое) вычисление.

Параметры:

• func (Callable) – Функция Python, принимающая один или несколько аргументов. Должна возвращать тензор из одного элемента. Если указано, что has_aux равно True, функция может возвращать кортеж из тензора из одного элемента и других вспомогательных объектов: (output, aux).
• argnums (int или Tuple[int]) – Указывает аргументы, относительно которых вычисляются градиенты. argnums может быть целым числом или кортежем целых чисел. Значение по умолчанию: 0.
• has_aux (bool) – Флаг, указывающий, что func возвращает тензор и другие вспомогательные объекты: (output, aux). Значение по умолчанию: False.

Возвращает:

Функцию для вычисления кортежа градиентов относительно её входных данных и прямого вычисления. По умолчанию результатом функции является кортеж, содержащий тензор (или тензоры) градиента относительно первого аргумента и результат прямого вычисления. Если указано, что has_aux равно True, возвращается кортеж градиентов и кортеж результатов прямого вычисления с выходными вспомогательными объектами. Если argnums является кортежем целых чисел, возвращается кортеж кортежей выходных градиентов относительно каждого значения argnums и результат прямого вычисления.

Тип возвращаемого значения:

Callable[[~_P], tuple[Any, Any]]

Примеры см. в разделе grad()