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

pub fn binary<B, const D_OUT: usize, const D_LHS: usize, const D_RHS: usize, 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<D_OUT>) -> <B as Backend>::FloatTensorPrimitive<D_LHS>,
    FRhs: FnOnce(<B as Backend>::FloatTensorPrimitive<D_OUT>) -> <B as Backend>::FloatTensorPrimitive<D_RHS>,
Expand description

Execute a binary operation during the backward step.