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

См. grad() для примеров