Function burn::backend::autodiff::ops::binary

pub fn binary<B, FLhs, FRhs>(
    parents: [Option<Arc<Node>>; 2],
    node: Arc<Node>,
    grads: &mut Gradients,
    func_lhs: FLhs,
    func_rhs: FRhs,
)
where
    B: Backend,
    FLhs: FnOnce(<B as Backend>::FloatTensorPrimitive) -> <B as Backend>::FloatTensorPrimitive,
    FRhs: FnOnce(<B as Backend>::FloatTensorPrimitive) -> <B as Backend>::FloatTensorPrimitive,
Expand description

Execute a binary operation during the backward step.