Improve float precision stability of linspace op, fix 4419. (#4470)

sighingnow · soumith · commit c43b120d4329 · 2018-01-03T22:45:26.000-05:00
Signed-off-by: HE, Tao &lt;sighingnow@gmail.com&gt;
diff --git a/aten/src/TH/generic/THTensorMath.c b/aten/src/TH/generic/THTensorMath.c
@@ -3383,7 +3383,7 @@ void THTensor_(linspace)(THTensor *r_, real a, real b, int64_t n)
     THTensor_(set1d)(r_, 0, a);
   } else {
      TH_TENSOR_APPLY(real, r_,
-             *r__data = a + i*(b-a)/((real)(n-1));
+             *r__data = a + (b-a)/((real)(n-1))*i;
              i++;
            );
   }

Original file line number	Diff line number	Diff line change
`@@ -3383,7 +3383,7 @@ void THTensor_(linspace)(THTensor *r_, real a, real b, int64_t n)`
`3383`	`3383`	`THTensor_(set1d)(r_, 0, a);`
`3384`	`3384`	`} else {`
`3385`	`3385`	`TH_TENSOR_APPLY(real, r_,`
`3386`		`- r__data = a + i(b-a)/((real)(n-1));`
	`3386`	`+ r__data = a + (b-a)/((real)(n-1))i;`
`3387`	`3387`	`i++;`
`3388`	`3388`	`);`
`3389`	`3389`	`}`