/usr/include/TH/THMath.h is in libtorch-th-dev 0~20170926-g89ede3b-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | #ifndef _THMATH_H
#define _THMATH_H
static inline double TH_sigmoid(double value) {
return 1.0 / (1.0 + exp(-value));
}
static inline double TH_frac(double x) {
return x - trunc(x);
}
static inline double TH_rsqrt(double x) {
return 1.0 / sqrt(x);
}
static inline double TH_lerp(double a, double b, double weight) {
return a + weight * (b-a);
}
static inline float TH_sigmoidf(float value) {
return 1.0f / (1.0f + expf(-value));
}
static inline float TH_fracf(float x) {
return x - truncf(x);
}
static inline float TH_rsqrtf(float x) {
return 1.0f / sqrtf(x);
}
static inline float TH_lerpf(float a, float b, float weight) {
return a + weight * (b-a);
}
#endif // _THMATH_H
|