| #ifndef TH_GENERIC_FILE |
| #define TH_GENERIC_FILE "generic/THTensorRandom.h" |
| #else |
| |
| TH_API void THTensor_(random)(THTensor *self, THGenerator *_generator); |
| TH_API void THTensor_(geometric)(THTensor *self, THGenerator *_generator, double p); |
| TH_API void THTensor_(bernoulli)(THTensor *self, THGenerator *_generator, double p); |
| TH_API void THTensor_(bernoulli_FloatTensor)(THTensor *self, THGenerator *_generator, THFloatTensor *p); |
| TH_API void THTensor_(bernoulli_DoubleTensor)(THTensor *self, THGenerator *_generator, THDoubleTensor *p); |
| |
| #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_(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); |
| TH_API void THTensor_(multinomial)(THLongTensor *self, THGenerator *_generator, THTensor *prob_dist, int n_sample, int with_replacement); |
| #endif |
| |
| #if defined(TH_REAL_IS_BYTE) |
| TH_API void THTensor_(getRNGState)(THGenerator *_generator, THTensor *self); |
| TH_API void THTensor_(setRNGState)(THGenerator *_generator, THTensor *self); |
| #endif |
| |
| #endif |
