Caffe2 - C++ API
A deep learning, cross platform ML framework
THCTensorRandom.h
1 #ifndef THC_GENERIC_FILE
2 #define THC_GENERIC_FILE "THC/generic/THCTensorRandom.h"
3 #else
4 
5 #if defined(THC_REAL_IS_FLOAT) || defined(THC_REAL_IS_DOUBLE) || defined(THC_REAL_IS_HALF)
6 
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);
18 
19 #endif
20 
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);
25 
26 #endif