grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
gradCheck
grain
testing
gradient check function to compare numeric grad and autograd
auto
gradCheck
(
F
In
Out
string
file
= __FILE__
size_t
line
= __LINE__
)
(
ref
F
func
,
In
inputs
,
Out
gradOutputs
,
float
eps
= 1e-3
,
float
rtol
= 1e-3
,
float
atol
= 1e-5
)
Meta
Source
See Implementation
grain
testing
functions
gradCheck
gradCheckChain
numericGrad
numericGradChain
gradient check function to compare numeric grad and autograd