grad_matmul

Function grad_matmul 

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

Matrix multiplication with gradient tracking.