diff options
| author | Soumith Chintala <soumith@gmail.com> | 2017-07-04 23:45:53 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2017-07-04 23:46:04 +0300 |
| commit | 0ea5868b68cb12f5d8d3e61a1237b46711c68a84 (patch) | |
| tree | acff5d0e0beefa7fb1ff2ddabfe9dac07fc11bda | |
| parent | a5fc2643fdf2cb20e140eeb09e5dd811c85c6ee6 (diff) | |
add missing definition
| -rw-r--r-- | lib/THC/generic/THCTensorMathReduce.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/lib/THC/generic/THCTensorMathReduce.h b/lib/THC/generic/THCTensorMathReduce.h index 33aee0f..3202a09 100644 --- a/lib/THC/generic/THCTensorMathReduce.h +++ b/lib/THC/generic/THCTensorMathReduce.h @@ -36,6 +36,11 @@ THC_API real THCTensor_(minall)(THCState *state, THCTensor *self); THC_API real THCTensor_(maxall)(THCState *state, THCTensor *self); THC_API real THCTensor_(medianall)(THCState *state, THCTensor *self); +THC_API void THCTensor_(median)(THCState *state, + THCTensor *values, + THCudaLongTensor *indices, + THCTensor *src, long dim, int keepdim); + THC_API accreal THCTensor_(dist)(THCState *state, THCTensor *self, THCTensor *src, real value); |