grad_mean

Function grad_mean 

Source
pub fn grad_mean(a: &GradientTensor) -> CoreResult<GradientTensor>
Expand description

Mean reduction with gradient tracking.