Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Multiprocessing package - tensorplay.multiprocessing Quantization tensorplay tensorplay.audio tensorplay.compiler tensorplay.cuda tensorplay.fft tensorplay.futures tensorplay.hub tensorplay.library tensorplay.linalg tensorplay.nn tensorplay.nn.functional tensorplay.nn.init tensorplay.optim tensorplay.random tensorplay.sparse tensorplay.special tensorplay.stax tensorplay.utils.checkpoint tensorplay.utils.data tensorplay.vision Upstream anchor labels

detect_anomaly enable_grad Function inference_mode no_grad set_detect_anomaly set_grad_enabled tensorplay.autograd.backward tensorplay.autograd.function.once_differentiable tensorplay.autograd.functional.hessian tensorplay.autograd.functional.hvp tensorplay.autograd.functional.jacobian tensorplay.autograd.functional.jvp tensorplay.autograd.functional.vhp tensorplay.autograd.functional.vjp tensorplay.autograd.grad tensorplay.autograd.grad_mode.is_grad_enabled tensorplay.autograd.gradcheck.get_analytical_jacobian tensorplay.autograd.gradcheck.get_numerical_jacobian tensorplay.autograd.gradcheck.get_numerical_jacobian_wrt_specific_input tensorplay.autograd.gradcheck.gradcheck tensorplay.autograd.gradcheck.GradcheckError tensorplay.autograd.gradcheck.gradgradcheck tensorplay.autograd.jvp

API symbolsautograd

Copy

View Markdown Download .md

tensorplay.autograd.gradcheck.get_numerical_jacobian_wrt_specific_input

tensorplay.autograd.gradcheck.get_numerical_jacobian_wrt_specific_input(fn, input_idx, inputs, outputs, eps, input=None) → tuple[source]

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

tensorplay.autograd.gradcheck.get_numerical_jacobian

Compute the numerical Jacobian for a given fn and its inputs.

tensorplay.autograd.gradcheck.gradcheck

Check gradients computed via small finite differences against analytical gradients wrt tensors in inputs that are of floating point or complex type and with requires_grad=True .