jvp

ivy.jvp(func, primals, tangents)

Compute a (forward-mode) Jacobian-vector product of func.

Parameters:

• func (callable) – Function to be differentiated.

• primals – sequence of primal values at which the Jacobian of func should be evaluated.

• tangents – sequence of tangent vectors giving the Jacobian-vector product of func evaluated at primals.

Returns:

ret – The output of func evaluated at primals. And the Jacobian-vector product of function evaluated at primals with tangents.