Skip to content

Commit af7aea9

Browse files
committed
Merge commit 'f805a8388be8dc55af0e3aa165b13cd0fce484d3'
2 parents 366299f + f805a83 commit af7aea9

File tree

2 files changed

+12
-12
lines changed

2 files changed

+12
-12
lines changed

torch/lib/TH/generic/THTensorMath.c

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

31503150
void THTensor_(linspace)(THTensor *r_, real a, real b, long n)

torch/lib/TH/generic/THTensorMath.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -169,17 +169,17 @@ TH_API void THTensor_(frac)(THTensor *r_, THTensor *t);
169169
TH_API void THTensor_(lerp)(THTensor *r_, THTensor *a, THTensor *b, real weight);
170170

171171
TH_API void THTensor_(mean)(THTensor *r_, THTensor *t, int dimension, int keepdim);
172-
TH_API void THTensor_(std)(THTensor *r_, THTensor *t, int dimension, int flag, int keepdim);
173-
TH_API void THTensor_(var)(THTensor *r_, THTensor *t, int dimension, int flag, int keepdim);
172+
TH_API void THTensor_(std)(THTensor *r_, THTensor *t, int dimension, int biased, int keepdim);
173+
TH_API void THTensor_(var)(THTensor *r_, THTensor *t, int dimension, int biased, int keepdim);
174174
TH_API void THTensor_(norm)(THTensor *r_, THTensor *t, real value, int dimension, int keepdim);
175175
TH_API void THTensor_(renorm)(THTensor *r_, THTensor *t, real value, int dimension, real maxnorm);
176176
TH_API accreal THTensor_(dist)(THTensor *a, THTensor *b, real value);
177177
TH_API void THTensor_(histc)(THTensor *hist, THTensor *tensor, long nbins, real minvalue, real maxvalue);
178178
TH_API void THTensor_(bhistc)(THTensor *hist, THTensor *tensor, long nbins, real minvalue, real maxvalue);
179179

180180
TH_API accreal THTensor_(meanall)(THTensor *self);
181-
TH_API accreal THTensor_(varall)(THTensor *self);
182-
TH_API accreal THTensor_(stdall)(THTensor *self);
181+
TH_API accreal THTensor_(varall)(THTensor *self, int biased);
182+
TH_API accreal THTensor_(stdall)(THTensor *self, int biased);
183183
TH_API accreal THTensor_(normall)(THTensor *t, real value);
184184

185185
TH_API void THTensor_(linspace)(THTensor *r_, real a, real b, long n);

0 commit comments

Comments
 (0)