fix/native: Fix sigmoid_grad to use x_diff instead of x for dx

autumnai · Jan 20, 2016 · c25a32a · c25a32a
1 parent 892ce8f
commit c25a32a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/frameworks/native/helper.rs b/src/frameworks/native/helper.rs
@@ -144,7 +144,7 @@ macro_rules! impl_ops_sigmoid_for {
                 result_diff: &mut ::co::tensor::SharedTensor<$t>
             ) -> Result<(), ::co::error::Error> {
                 if let Some(sig_data) = x.get(self.device()).unwrap().as_native() {
-                    if let Some(sig_dx) = x.get(self.device()).unwrap().as_native() {
+                    if let Some(sig_dx) = x_diff.get(self.device()).unwrap().as_native() {
                         let res = sig_data.as_slice::<$t>().iter()
                         .zip(sig_dx.as_slice::<$t>().iter())
                         .map(|(t, dt)| ::frameworks::native::helper::sigmoid_grad(t, dt));