| commit | c94f3bbf339f4378a97c12e2c1e3627cce8288ba | [log] [tgz] |
|---|---|---|
| author | James Reed <jamesreed@fb.com> | Wed May 30 15:11:24 2018 -0700 |
| committer | Soumith Chintala <soumith@gmail.com> | Wed May 30 18:11:24 2018 -0400 |
| tree | baccb1488875b91dffa914d9b187739f3a82dd4b | |
| parent | 2e78bfa530c3a1bd4d23e23237dd8c33522955f7 [diff] |
Fix typo in autodiff formula for addmm (#7932)
diff --git a/torch/csrc/jit/autodiff.cpp b/torch/csrc/jit/autodiff.cpp index 2658895..3c9077b 100644 --- a/torch/csrc/jit/autodiff.cpp +++ b/torch/csrc/jit/autodiff.cpp
@@ -95,7 +95,7 @@ dmat2 = inputs.at(0).t().mm(grads.at(0)); } } else { - dmat2 = grads.at(0).mm(inputs.at(1).t()); + dmat2 = inputs.at(0).t().mm(grads.at(0)); } return {dmat1, dmat2}; }