Skip to content

Commit 295ed7e

Browse files
committed
Merge commit 'ab7d4e2bcea5cae8f05873fb0bbb31985cc58d47'
2 parents ae65236 + ab7d4e2 commit 295ed7e

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

torch/lib/THC/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)