grad_multiply

Function grad_multiply 

Source
pub fn grad_multiply(
    a: &GradientTensor,
    b: &GradientTensor,
) -> CoreResult<GradientTensor>
Expand description

Element-wise multiplication with gradient tracking.