Function auto_diff::op::_gradient_checker
source · [−]pub fn _gradient_checker(
op: &mut dyn OpTrait,
one_input: &[&Tensor],
input_mask: Option<&[bool]>,
step: Option<f32>,
tolerance: Option<f32>
) -> boolExpand description
Verify the gradient implementation is right.