jax.test_util 模块

函数列表

check_grads(f, args, order[, modes, atol, ...])	使用有限差分检查自动微分的梯度。
check_jvp(f, f_jvp, args[, atol, rtol, eps, ...])	使用有限差分检查自动微分的 JVP。
check_vjp(f, f_vjp, args[, atol, rtol, eps, ...])	使用有限差分检查自动微分的 VJP。