diff options
Diffstat (limited to 'lib/THC/generic/THCTensorMath.cu')
| -rw-r--r-- | lib/THC/generic/THCTensorMath.cu | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/lib/THC/generic/THCTensorMath.cu b/lib/THC/generic/THCTensorMath.cu index a0e550a..9ffc89b 100644 --- a/lib/THC/generic/THCTensorMath.cu +++ b/lib/THC/generic/THCTensorMath.cu @@ -36,6 +36,14 @@ THCTensor_(zero)(THCState *state, THCTensor *self_) } THC_API void +THCTensor_(mean)(THCState *state, THCudaTensor *self, THCudaTensor *src, long dim) +{ + THAssert(THCTensor_(checkGPU)(state, 2, self, src)); + THCudaTensor_sum(state, self, src, dim); + THCudaTensor_div(state, self, self, THCudaTensor_size(state, src, dim)); +} + +THC_API void THCTensor_(zeros)(THCState *state, THCTensor *r_, THLongStorage *size) { THAssert(THCTensor_(checkGPU)(state, 1, r_)); |