diff options
| author | Adam Paszke <adam.paszke@gmail.com> | 2016-09-14 17:32:37 +0300 |
|---|---|---|
| committer | Adam Paszke <adam.paszke@gmail.com> | 2016-09-14 17:32:37 +0300 |
| commit | 186b06492d9bfd2c4fa3302e8d4d1d46a90a711a (patch) | |
| tree | 352bd2be367af9869bb4af8c09099d295c34b24d | |
| parent | 8f2b88a5a4f2b541b9efc2fb72a47572811efd17 (diff) | |
Mark BCECriterion weights as optional in THNN.h
| -rw-r--r-- | lib/THNN/generic/THNN.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/lib/THNN/generic/THNN.h b/lib/THNN/generic/THNN.h index 0f2149a..f6ff7cc 100644 --- a/lib/THNN/generic/THNN.h +++ b/lib/THNN/generic/THNN.h @@ -31,14 +31,14 @@ TH_API void THNN_(BCECriterion_updateOutput)( THTensor *target, THTensor *output, bool sizeAverage, - THTensor *weights); + THTensor *weights); // [OPTIONAL] TH_API void THNN_(BCECriterion_updateGradInput)( THNNState *state, THTensor *input, THTensor *target, THTensor *gradInput, bool sizeAverage, - THTensor *weights); + THTensor *weights); // [OPTIONAL] TH_API void THNN_(ClassNLLCriterion_updateOutput)( THNNState *state, // library's state |