diff options
| author | txh18 <cloudygooseg@gmail.com> | 2015-11-25 18:42:26 +0800 |
|---|---|---|
| committer | txh18 <cloudygooseg@gmail.com> | 2015-11-25 18:42:26 +0800 |
| commit | ca3500f01ea7ce695a4dbf70d2be8244827097c9 (patch) | |
| tree | 5fa85c778c16a40279cd2bb331f8511aae2b5dca /nerv/lib/matrix/generic/cumatrix.h | |
| parent | 8e590ba284bfee414659f1845e175b41cac05d45 (diff) | |
added tanh operation for matrix
Diffstat (limited to 'nerv/lib/matrix/generic/cumatrix.h')
| -rw-r--r-- | nerv/lib/matrix/generic/cumatrix.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/nerv/lib/matrix/generic/cumatrix.h b/nerv/lib/matrix/generic/cumatrix.h index 21c29b7..aa8805a 100644 --- a/nerv/lib/matrix/generic/cumatrix.h +++ b/nerv/lib/matrix/generic/cumatrix.h @@ -9,6 +9,9 @@ void nerv_matrix_(mul)(Matrix *c, const Matrix *a, const Matrix *b, void nerv_matrix_(sigmoid)(Matrix *a, const Matrix *b, Status *status); void nerv_matrix_(sigmoid_grad)(Matrix *nerr, const Matrix *err, const Matrix *output, Status *status); +void nerv_matrix_(tanh)(Matrix *a, const Matrix *b, Status *status); +void nerv_matrix_(tanh_grad)(Matrix *nerr, const Matrix *err, + const Matrix *output, Status *status); Matrix *nerv_matrix_(softmax)(Matrix *b, const Matrix *a, Status *status); Matrix *nerv_matrix_(rowsum)(Matrix *a, Status *status); |