pub fn grad_matmul(
a: &GradientTensor,
b: &GradientTensor,
) -> CoreResult<GradientTensor>
Expand description
Matrix multiplication with gradient tracking.
pub fn grad_matmul(
a: &GradientTensor,
b: &GradientTensor,
) -> CoreResult<GradientTensor>
Matrix multiplication with gradient tracking.