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