lysergic
Version:
Synaptic's neural network compiler
29 lines (28 loc) • 758 B
TypeScript
import { ExpressionNode, Variable } from "./nodes";
export declare const WHOLE_LAYER_ACTIVATION_KIND = 128;
export declare enum ActivationTypes {
IDENTITY = 0,
LOGISTIC_SIGMOID = 1,
TANH = 2,
INVERSE_IDENTITY = 5,
EXP = 6,
SOFTPLUS = 7,
SOFTSIGN = 8,
GAUSSIAN = 9,
STEP = 11,
RELU = 12,
PRELU = 13,
RELU_PLUSONE = 14,
ELU = 15,
PELU = 16,
POW = 17,
POW_MINUS1 = 18,
AVG_POOLING,
MAX_POOLING,
MAXOUT,
SOFTMAX,
SHARPEN,
BATCH_NORM,
}
export declare function buildActivationFunction(state: ExpressionNode, type: ActivationTypes): ExpressionNode;
export declare function buildDerivativeFunction(state: Variable, activation: Variable, type: ActivationTypes): ExpressionNode;