diff options
| author | Andreas Köpf <andreas.koepf@xamla.com> | 2016-01-03 00:04:29 +0300 |
|---|---|---|
| committer | Andreas Köpf <andreas.koepf@xamla.com> | 2016-01-03 00:04:29 +0300 |
| commit | 0b73786475c75d946bd5e5c7880a488a50f43c38 (patch) | |
| tree | 05d53044f15f3e87435d02929eccdcb023dedb7a /THNN.lua | |
| parent | b0d35fe27965b1028075e21c1099debb77f9cbc2 (diff) | |
Use tensor for THNN functions even for single element outputs
Diffstat (limited to 'THNN.lua')
| -rw-r--r-- | THNN.lua | 6 |
1 files changed, 3 insertions, 3 deletions
@@ -17,7 +17,7 @@ TH_API void THNN_(AbsCriterion_updateOutput)( THNNState *state, THTensor *input, THTensor *target, - real *output, + THTensor *output, bool sizeAverage); TH_API void THNN_(AbsCriterion_updateGradInput)( THNNState *state, @@ -47,7 +47,7 @@ TH_API void THNN_(DistKLDivCriterion_updateOutput)( THNNState *state, THTensor *input, THTensor *target, - real *output, + THTensor *output, bool sizeAverage); TH_API void THNN_(DistKLDivCriterion_updateGradInput)( THNNState *state, @@ -85,7 +85,7 @@ TH_API void THNN_(HardTanh_updateGradInput)( TH_API void THNN_(L1Cost_updateOutput)( THNNState *state, THTensor *input, - real *output); + THTensor *output); TH_API void THNN_(L1Cost_updateGradInput)( THNNState *state, THTensor *input, |