Skip to content

Commit ab7d4e2

Browse files
committed
add missing definition
1 parent 0dbf871 commit ab7d4e2

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

generic/THCTensorMathReduce.h

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,11 @@ THC_API real THCTensor_(minall)(THCState *state, THCTensor *self);
3636
THC_API real THCTensor_(maxall)(THCState *state, THCTensor *self);
3737
THC_API real THCTensor_(medianall)(THCState *state, THCTensor *self);
3838

39+
THC_API void THCTensor_(median)(THCState *state,
40+
THCTensor *values,
41+
THCudaLongTensor *indices,
42+
THCTensor *src, long dim, int keepdim);
43+
3944
THC_API accreal THCTensor_(dist)(THCState *state, THCTensor *self, THCTensor *src,
4045
real value);
4146

0 commit comments

Comments
 (0)