fn scale()

in gad/src/array.rs [309:329]

• 21 lines of code
• 3 McCabe index (conditional complexity)

            fn scale(&mut self, v1: &Value<T>, v2: &Value<D>) -> Value<D> {
                let result = self.eval().scale(v1.data(), v2.data());
                let value = self.make_node(result, vec![v1.input(), v2.input()], {
                    let v1 = v1.clone();
                    let v2 = v2.clone();
                    move |graph, store, gradient| {
                        if let Some(id) = v1.id() {
                            let c2 = graph.link(&v2);
                            let grad = graph.dot(&gradient, c2)?;
                            store.add_gradient::<T, _>(graph, id, &grad)?;
                        }
                        if let Some(id) = v2.id() {
                            let c1 = graph.link(&v1);
                            let grad = graph.scale(c1, &gradient);
                            store.add_gradient::<D, _>(graph, id, &grad)?;
                        }
                        Ok(())
                    }
                });
                value
            }