lab.autograd.custom module

lab.autograd.custom.autograd_register(f, s_f)

Register a function and its sensitivity for AutoGrad.

Parameters

• f (function) – Function to register.

• s_f (function) – Sensitivity of f.

Returns

AutoGrad primitive.

Return type

function