// TODO: move this somewhere - we only need one version
static std::string THPStorage_(__newHandle)() {
std::string handle = "/torch_";
#ifdef _MSC_VER
handle += std::to_string(GetCurrentProcessId());
#else
handle += std::to_string(getpid());
#endif
handle += "_";
handle += std::to_string(THRandom_random(THPGenerator_TH_CData(THPDefaultGenerator)));
return handle;
}
#ifndef TH_GENERIC_FILE
#define TH_GENERIC_FILE "generic/THTensorRandom.c"
#else
TH_API void THTensor_(random)(THTensor *self)
{
#if defined(TH_REAL_IS_BYTE)
TH_TENSOR_APPLY(real, self, *self_data = (unsigned char)(THRandom_random() % (UCHAR_MAX+1)););
#elif defined(TH_REAL_IS_CHAR)
TH_TENSOR_APPLY(real, self, *self_data = (char)(THRandom_random() % (CHAR_MAX+1)););
#elif defined(TH_REAL_IS_SHORT)
TH_TENSOR_APPLY(real, self, *self_data = (short)(THRandom_random() % (SHRT_MAX+1)););
#elif defined(TH_REAL_IS_INT)
TH_TENSOR_APPLY(real, self, *self_data = (int)(THRandom_random() % (INT_MAX+1UL)););
#elif defined(TH_REAL_IS_LONG)
TH_TENSOR_APPLY(real, self, *self_data = (long)(THRandom_random() % (LONG_MAX+1UL)););
#elif defined(TH_REAL_IS_FLOAT)
TH_TENSOR_APPLY(real, self, *self_data = (float)(THRandom_random() % ((1UL << FLT_MANT_DIG)+1)););
#elif defined(TH_REAL_IS_DOUBLE)
TH_TENSOR_APPLY(real, self, *self_data = (double)(THRandom_random() % ((1ULL << DBL_MANT_DIG)+1)););
#else
#error "Unknown type"
#endif
}
TH_API void THTensor_(geometric)(THTensor *self, double p)
{
TH_TENSOR_APPLY(real, self, *self_data = (real)THRandom_geometric(p););
}
TH_API void THTensor_(bernoulli)(THTensor *self, double p)
