diff options
author | Trevor Killeen <killeent@devgpu047.ash3.facebook.com> | 2016-09-26 22:32:08 +0300 |
---|---|---|
committer | Trevor Killeen <killeentm@gmail.com> | 2016-10-07 21:50:27 +0300 |
commit | 72123175666a3860a093fc7fa5ea26628b89d379 (patch) | |
tree | 3466d93ed5809f904448e7b8dac52a563c51d048 /lib/THC/generic/THCTensorMath.h | |
parent | d385fb68798d4f5258980a9b3c4ccbc75faeacfe (diff) |
[cutorch refactor] move mean function into generic/
Diffstat (limited to 'lib/THC/generic/THCTensorMath.h')
-rw-r--r-- | lib/THC/generic/THCTensorMath.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/lib/THC/generic/THCTensorMath.h b/lib/THC/generic/THCTensorMath.h index 5c9e66d..6b59262 100644 --- a/lib/THC/generic/THCTensorMath.h +++ b/lib/THC/generic/THCTensorMath.h @@ -4,6 +4,7 @@ THC_API void THCTensor_(fill)(THCState *state, THCTensor *self, real value); THC_API void THCTensor_(zero)(THCState *state, THCTensor *self); +THC_API void THCTensor_(mean)(THCState *state, THCudaTensor *self, THCudaTensor *src, long dim); THC_API void THCTensor_(zeros)(THCState *state, THCTensor *r_, THLongStorage *size); THC_API void THCTensor_(ones)(THCState *state, THCTensor *r_, THLongStorage *size); |