burn-autodiff 0.20.0

Automatic differentiation backend for the Burn framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

use alloc::vec::Vec;

use crate::graph::NodeRef;
/// Duplicate the given object for each node that requires gradients.
///
/// # Notes
///
/// This is useful since you don't have to keep N cloned references alive event if just 1 node
/// will be updated.
///
/// If the object is a tensor and if one reference exists, it can be updated inplace.
pub fn duplicate<T: Clone + core::fmt::Debug, const N: usize>(
    nodes: &[Option<NodeRef>; N],
    obj: Option<T>,
) -> [Option<T>; N] {
    nodes
        .iter()
        .map(|node| match node {
            Some(_) => obj.clone(),
            None => None,
        })
        .collect::<Vec<_>>()
        .try_into()
        .unwrap()
}