diff options
Diffstat (limited to 'lib/TH/generic/THTensorMath.h')
-rw-r--r-- | lib/TH/generic/THTensorMath.h | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/lib/TH/generic/THTensorMath.h b/lib/TH/generic/THTensorMath.h index 87f1616..781318c 100644 --- a/lib/TH/generic/THTensorMath.h +++ b/lib/TH/generic/THTensorMath.h @@ -128,6 +128,16 @@ TH_API void THTensor_(eqTensorT)(THTensor *r_, THTensor *ta, THTensor *tb); TH_API void THTensor_(abs)(THTensor *r_, THTensor *t); #endif +#if defined(TH_REAL_IS_FLOAT) || defined(TH_REAL_IS_DOUBLE) || defined(TH_REAL_IS_HALF) +TH_API void THTensor_(rand)(THTensor *r_, THGenerator *_generator, THLongStorage *size); +TH_API void THTensor_(randn)(THTensor *r_, THGenerator *_generator, THLongStorage *size); +#endif + +#if defined(TH_REAL_IS_FLOAT) || defined(TH_REAL_IS_DOUBLE) +TH_API void THTensor_(rand)(THTensor *r_, THGenerator *_generator, THLongStorage *size); +TH_API void THTensor_(randn)(THTensor *r_, THGenerator *_generator, THLongStorage *size); +#endif + #if defined(TH_REAL_IS_FLOAT) || defined(TH_REAL_IS_DOUBLE) TH_API void THTensor_(sigmoid)(THTensor *r_, THTensor *t); @@ -171,9 +181,6 @@ TH_API accreal THTensor_(normall)(THTensor *t, real value); TH_API void THTensor_(linspace)(THTensor *r_, real a, real b, long n); TH_API void THTensor_(logspace)(THTensor *r_, real a, real b, long n); -TH_API void THTensor_(rand)(THTensor *r_, THGenerator *_generator, THLongStorage *size); -TH_API void THTensor_(randn)(THTensor *r_, THGenerator *_generator, THLongStorage *size); - #endif #if defined(TH_REAL_IS_BYTE) |