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