diff options
| author | Qi Liu <liuq901@163.com> | 2016-04-29 16:00:20 +0800 |
|---|---|---|
| committer | Qi Liu <liuq901@163.com> | 2016-04-29 16:00:20 +0800 |
| commit | 1f5568a9e1457dcf5aadd08749aef6194370b43f (patch) | |
| tree | d705bbec5b701774a144810c5e60a73e346f9c39 /nerv/lib/matrix/generic/mmatrix.h | |
| parent | 6051c37a2c55365a5834d5b4c11d973e2e4e1ad4 (diff) | |
add relu layer
Diffstat (limited to 'nerv/lib/matrix/generic/mmatrix.h')
| -rw-r--r-- | nerv/lib/matrix/generic/mmatrix.h | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/nerv/lib/matrix/generic/mmatrix.h b/nerv/lib/matrix/generic/mmatrix.h index 41c39f6..7f494d6 100644 --- a/nerv/lib/matrix/generic/mmatrix.h +++ b/nerv/lib/matrix/generic/mmatrix.h @@ -13,6 +13,16 @@ void nerv_matrix_(sigmoid)(Matrix *a, const Matrix *b, void nerv_matrix_(sigmoid_grad)(Matrix *nerr, const Matrix *err, const Matrix *output, MContext *context, Status *status); +void nerv_matrix_(tanh)(Matrix *a, const Matrix *b, + MContext *context, Status *status); +void nerv_matrix_(tanh_grad)(Matrix *nerr, const Matrix *err, + const Matrix *output, + MContext *context, Status *status); +void nerv_matrix_(relu)(Matrix *a, const Matrix *b, + MContext *context, Status *status); +void nerv_matrix_(relu_grad)(Matrix *nerr, const Matrix *err, + const Matrix *output, + MContext *context, Status *status); Matrix *nerv_matrix_(softmax)(Matrix *b, const Matrix *a, MContext *context, Status *status); |