PlusML
Loading...
Searching...
No Matches
util.h
1#ifndef UTIL_H
2#define UTIL_H
3
4#define EXPORT __declspec(dllexport)
5#include <Eigen/Dense>
6
7namespace plusml {
8EXPORT float Sign(float n);
9EXPORT double Sign(double n);
10
11EXPORT float Sigmoid(float n);
12EXPORT double Sigmoid(double n);
13
14EXPORT Eigen::VectorXf Softmax(const Eigen::VectorXf& vec);
15EXPORT void SoftmaxInPlace(Eigen::VectorXf& vec);
16
17EXPORT uint64_t Argmax(const Eigen::VectorXf& vec);
18} //namespace plusml
19
20#endif //UTIL_H