There was an error while loading. Please reload this page.
1 parent 05e4a1a commit c98e30eCopy full SHA for c98e30e
1 file changed
stan/math/rev/fun/squared_distance.hpp
@@ -21,7 +21,8 @@ namespace math {
21
inline var squared_distance(const var& a, const var& b) {
22
check_finite("squared_distance", "a", a);
23
check_finite("squared_distance", "b", b);
24
- return make_callback_vari((a.val() - b.val()) * (a.val() - b.val()),
+ double difference = a.val() - b.val();
25
+ return make_callback_vari(difference * difference,
26
[a, b](const auto& vi) mutable {
27
const double diff = 2.0 * (a.val() - b.val());
28
a.adj() += vi.adj_ * diff;
@@ -35,7 +36,8 @@ inline var squared_distance(const var& a, const var& b) {
35
36
inline var squared_distance(const var& a, double b) {
37
38
- return make_callback_vari((a.val() - b) * (a.val() - b),
39
+ double difference = a.val() - b;
40
41
42
a.adj() += vi.adj_ * 2.0 * (a.val() - b);
43
});
0 commit comments