Add reduceMin operation using #define

source/tensor/core/reduce/ReduceMax.cuh

@@ -31,6 +31,9 @@ namespace nts{ // namespace nts(NiuTrans.Tensor)
 /* get the max-valued items along a dimension of the tensor (cuda version) */
 /* get the max-valued items along a dimension of the tensor (cuda version) */
 void _CudaReduceMax(const XTensor * input, XTensor * output, int dim);
 void _CudaReduceMax(const XTensor * input, XTensor * output, int dim);
 
 
+/* get the min-valued items along a dimension of the tensor (cuda version) */
+void _CudaReduceMin(const XTensor * input, XTensor * output, int dim);
+
 #endif // USE_CUDA
 #endif // USE_CUDA
 
 
 } // namespace nts(NiuTrans.Tensor)
 } // namespace nts(NiuTrans.Tensor)

source/tensor/core/reduce/ReduceMax.h

@@ -29,12 +29,21 @@ namespace nts{ // namespace nts(NiuTrans.Tensor)
 /* get the max value of the items along a dimension of the tensor. */
 /* get the max value of the items along a dimension of the tensor. */
 void _ReduceMax(const XTensor * input, XTensor * output, int dim);
 void _ReduceMax(const XTensor * input, XTensor * output, int dim);
 
 
+/* get the min value of the items along a dimension of the tensor. */
+void _ReduceMin(const XTensor * input, XTensor * output, int dim);
+
 /* 
 /* 
 get the max value of the items along a dimension of the tensor (return an XTensor structure)
 get the max value of the items along a dimension of the tensor (return an XTensor structure)
 make a new tensor to keep the result and return it
 make a new tensor to keep the result and return it
 */
 */
 XTensor ReduceMax(const XTensor &input, int dim);
 XTensor ReduceMax(const XTensor &input, int dim);
 
 
+/*
+get the min value of the items along a dimension of the tensor (return an XTensor structure)
+make a new tensor to keep the result and return it
+*/
+XTensor ReduceMin(const XTensor &input, int dim);
+
 } // namespace nts(NiuTrans.Tensor)
 } // namespace nts(NiuTrans.Tensor)
 
 
 #endif // __REDUCEMAX_H__
 #endif // __REDUCEMAX_H__

source/tensor/core/reduce/VectorBuffer.cpp

@@ -168,4 +168,13 @@ VectorBuffer VectorBuffer::maxData(const VectorBuffer &a) {
     return *this;
     return *this;
 }
 }
 
 
+/* conculte the max of two buffer */
+VectorBuffer VectorBuffer::minData(const VectorBuffer &a) {
+    for (int i = 0; i != a.size(); i++) {
+        this->values[i] = MIN(a[i], this->values[i]);
+        printf("runhere");
+    }
+    return *this;
+}
+
 }/* end of the nts (NiuTrans.Tensor) namespace */
 }/* end of the nts (NiuTrans.Tensor) namespace */
\ No newline at end of file

source/tensor/core/reduce/VectorBuffer.h

@@ -48,5 +48,8 @@ public:
 
 
     /* conculte the max of two buffer */
     /* conculte the max of two buffer */
     VectorBuffer maxData(const VectorBuffer &a); 
     VectorBuffer maxData(const VectorBuffer &a); 
+
+    /* conculte the max of two buffer */
+    VectorBuffer minData(const VectorBuffer &a);
 };
 };
 }
 }
\ No newline at end of file