#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