diff options
| author | Trevor Killeen <killeentm@gmail.com> | 2017-08-04 21:32:52 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2017-08-05 02:49:30 +0300 |
| commit | 723cc811227aca9b89ca6e74db8d0871bf030dd2 (patch) | |
| tree | 80c60f4122f8e23e509de5065daf69e3ddaf47d2 | |
| parent | aed31711c6b8846b8337a263a7f9f998697994e7 (diff) | |
move normal variants to TH/THC
| -rw-r--r-- | lib/TH/generic/THTensorRandom.c | 23 | ||||
| -rw-r--r-- | lib/TH/generic/THTensorRandom.h | 3 |
2 files changed, 26 insertions, 0 deletions
diff --git a/lib/TH/generic/THTensorRandom.c b/lib/TH/generic/THTensorRandom.c index 595cfa7..fe1401c 100644 --- a/lib/TH/generic/THTensorRandom.c +++ b/lib/TH/generic/THTensorRandom.c @@ -55,6 +55,29 @@ void THTensor_(normal)(THTensor *self, THGenerator *_generator, double mean, dou TH_TENSOR_APPLY(real, self, *self_data = (real)THRandom_normal(_generator, mean, stdv);); } +void THTensor_(normal_means)(THTensor *self, THGenerator *gen, THTensor *means, double stddev) +{ + THTensor_(resizeAs)(self, means); + THTensor_(normal)(self, gen, 0, stddev); + THTensor_(cadd)(self, self, 1, means); +} + +void THTensor_(normal_stddevs)(THTensor *self, THGenerator *gen, double mean, THTensor *stddevs) +{ + THTensor_(resizeAs)(self, stddevs); + THTensor_(normal)(self, gen, 0, 1); + THTensor_(cmul)(self, self, stddevs); + THTensor_(add)(self, self, mean); +} + +void THTensor_(normal_means_stddevs)(THTensor *self, THGenerator *gen, THTensor *means, THTensor *stddevs) +{ + THTensor_(resizeAs)(self, means); + THTensor_(normal)(self, gen, 0, 1); + THTensor_(cmul)(self, self, stddevs); + THTensor_(cadd)(self, self, 1, means); +} + void THTensor_(exponential)(THTensor *self, THGenerator *_generator, double lambda) { TH_TENSOR_APPLY(real, self, *self_data = (real)THRandom_exponential(_generator, lambda);); diff --git a/lib/TH/generic/THTensorRandom.h b/lib/TH/generic/THTensorRandom.h index e39d589..f79a86e 100644 --- a/lib/TH/generic/THTensorRandom.h +++ b/lib/TH/generic/THTensorRandom.h @@ -11,6 +11,9 @@ TH_API void THTensor_(bernoulli_DoubleTensor)(THTensor *self, THGenerator *_gene #if defined(TH_REAL_IS_FLOAT) || defined(TH_REAL_IS_DOUBLE) TH_API void THTensor_(uniform)(THTensor *self, THGenerator *_generator, double a, double b); TH_API void THTensor_(normal)(THTensor *self, THGenerator *_generator, double mean, double stdv); +TH_API void THTensor_(normal_means)(THTensor *self, THGenerator *gen, THTensor *means, double stddev); +TH_API void THTensor_(normal_stddevs)(THTensor *self, THGenerator *gen, double mean, THTensor *stddevs); +TH_API void THTensor_(normal_means_stddevs)(THTensor *self, THGenerator *gen, THTensor *means, THTensor *stddevs); TH_API void THTensor_(exponential)(THTensor *self, THGenerator *_generator, double lambda); TH_API void THTensor_(cauchy)(THTensor *self, THGenerator *_generator, double median, double sigma); TH_API void THTensor_(logNormal)(THTensor *self, THGenerator *_generator, double mean, double stdv); |