@@ -2925,7 +2925,7 @@ void THTensor_(mean)(THTensor *r_, THTensor *t, int dimension, int keepdim)
29252925 THTensor_ (div )(r_ , r_ , t -> size [dimension ]);
29262926}
29272927
2928- void THTensor_ (std )(THTensor * r_ , THTensor * t , int dimension , int flag , int keepdim )
2928+ void THTensor_ (std )(THTensor * r_ , THTensor * t , int dimension , int biased , int keepdim )
29292929{
29302930 THLongStorage * dim ;
29312931
@@ -2948,7 +2948,7 @@ void THTensor_(std)(THTensor *r_, THTensor *t, int dimension, int flag, int keep
29482948 sum2 += z * z ;
29492949 }
29502950
2951- if (flag )
2951+ if (biased )
29522952 {
29532953 sum /= t_size ;
29542954 sum2 /= t_size ;
@@ -2970,7 +2970,7 @@ void THTensor_(std)(THTensor *r_, THTensor *t, int dimension, int flag, int keep
29702970 }
29712971}
29722972
2973- void THTensor_ (var )(THTensor * r_ , THTensor * t , int dimension , int flag , int keepdim )
2973+ void THTensor_ (var )(THTensor * r_ , THTensor * t , int dimension , int biased , int keepdim )
29742974{
29752975 THLongStorage * dim ;
29762976
@@ -2993,7 +2993,7 @@ void THTensor_(var)(THTensor *r_, THTensor *t, int dimension, int flag, int keep
29932993 sum2 += z * z ;
29942994 }
29952995
2996- if (flag )
2996+ if (biased )
29972997 {
29982998 sum /= t_size ;
29992999 sum2 /= t_size ;
@@ -3133,18 +3133,18 @@ accreal THTensor_(meanall)(THTensor *tensor)
31333133 return THTensor_ (sumall )(tensor )/THTensor_ (nElement )(tensor );
31343134}
31353135
3136- accreal THTensor_ (varall )(THTensor * tensor )
3136+ accreal THTensor_ (varall )(THTensor * tensor , int biased )
31373137{
31383138 accreal mean = THTensor_ (meanall )(tensor );
31393139 accreal sum = 0 ;
31403140 TH_TENSOR_APPLY (real , tensor , sum += (* tensor_data - mean )* (* tensor_data - mean ););
3141- sum /= ( THTensor_ (nElement )(tensor )- 1 );
3141+ sum /= THTensor_ (nElement )(tensor ) - ( biased ? 0 : 1 );
31423142 return sum ;
31433143}
31443144
3145- accreal THTensor_ (stdall )(THTensor * tensor )
3145+ accreal THTensor_ (stdall )(THTensor * tensor , int biased )
31463146{
3147- return sqrt (THTensor_ (varall )(tensor ));
3147+ return sqrt (THTensor_ (varall )(tensor , biased ));
31483148}
31493149
31503150void THTensor_ (linspace )(THTensor * r_ , real a , real b , long n )
0 commit comments