1 #ifndef THC_GENERIC_FILE 2 #define THC_GENERIC_FILE "THC/generic/THCTensorRandom.h" 5 #if defined(THC_REAL_IS_FLOAT) || defined(THC_REAL_IS_DOUBLE) || defined(THC_REAL_IS_HALF) 7 THC_API
void THCTensor_(uniform)(
struct THCState *state, THCTensor *
self,
double a,
double b);
8 THC_API
void THCTensor_(normal)(
struct THCState *state, THCTensor *
self,
double mean,
double stdv);
9 THC_API
void THCTensor_(normal_means)(
struct THCState *state, THCTensor *
self, THCTensor *means,
double stddev);
10 THC_API
void THCTensor_(normal_stddevs)(
struct THCState *state, THCTensor *
self,
double mean, THCTensor *stddevs);
11 THC_API
void THCTensor_(normal_means_stddevs)(
struct THCState *state, THCTensor *
self, THCTensor *means, THCTensor *stddevs);
12 THC_API
void THCTensor_(logNormal)(
struct THCState *state, THCTensor *
self,
double mean,
double stdv);
13 THC_API
void THCTensor_(exponential)(
struct THCState *state, THCTensor *
self,
double lambda);
14 THC_API
void THCTensor_(cauchy)(
struct THCState *state, THCTensor *
self,
double median,
double sigma);
15 THC_API
void THCTensor_(multinomial)(
struct THCState *state, THCudaLongTensor *
self, THCTensor *prob_dist,
int n_sample,
int with_replacement);
16 THC_API
void THCTensor_(multinomialAliasSetup)(
struct THCState *state, THCTensor *probs, THCudaLongTensor *J, THCTensor *q);
17 THC_API
void THCTensor_(multinomialAliasDraw)(THCState *state, THCudaLongTensor *
self, THCudaLongTensor *_J, THCTensor *_q);
21 THC_API
void THCTensor_(random)(
struct THCState *state, THCTensor *
self);
22 THC_API
void THCTensor_(clampedRandom)(
struct THCState *state, THCTensor *
self, int64_t min, int64_t max);
23 THC_API
void THCTensor_(cappedRandom)(
struct THCState *state, THCTensor *
self, int64_t max);
24 THC_API
void THCTensor_(geometric)(
struct THCState *state, THCTensor *
self,
double p);