torch.func.grad_and_value¶

torch.func.grad_and_value(func, argnums=0, has_aux=False)¶

返回一个函数来计算梯度和原始或前向计算的元组。

Parameters

func (可调用对象) – 一个接受一个或多个参数的Python函数。必须返回一个单元素张量。如果指定 has_aux 等于 True，函数可以返回一个包含单元素张量和其他辅助对象的元组：(output, aux)。
argnums (整数或元组[整数]) – 指定要计算梯度的参数。argnums 可以是单个整数或整数元组。默认值：0。
has_aux (bool) – 标志指示 func 返回一个张量和其他辅助对象：(output, aux)。默认值：False。

Returns

计算相对于其输入和前向计算的梯度元组的函数。默认情况下，函数的输出是相对于第一个参数的梯度张量元组和原始计算。如果指定 has_aux 等于 True，则返回梯度元组和包含输出辅助对象的前向计算元组。如果 argnums 是一个整数元组，则返回一个元组，其中包含相对于每个 argnums 值的输出梯度元组和前向计算。

Return type

可调用

参见 grad() 的示例