Function burn::backend::autodiff::ops::unary

pub fn unary<B, const D_OUT: usize, const D_IN: usize, F>(
    parents: [Option<Arc<Node>>; 1],
    node: Arc<Node>,
    grads: &mut Gradients,
    func: F,
)
where
    B: Backend,
    F: FnOnce(<B as Backend>::FloatTensorPrimitive<D_OUT>) -> <B as Backend>::FloatTensorPrimitive<D_IN>,
Expand description

Execute a unary operation during the backward step.