pub fn grad_add(
a: &GradientTensor,
b: &GradientTensor,
) -> CoreResult<GradientTensor>
Expand description
Implementations of gradient-aware operations Addition operation with gradient tracking.
pub fn grad_add(
a: &GradientTensor,
b: &GradientTensor,
) -> CoreResult<GradientTensor>
Implementations of gradient-aware operations Addition operation with gradient tracking.