functions for float-typed data

source/tensor/core/math/Binary.h

@@ -27,24 +27,25 @@
 namespace nts { // namespace nts(NiuTrans.Tensor)
 namespace nts { // namespace nts(NiuTrans.Tensor)
 
 
 /*
 /*
-scale all tensor entires
+scale up tensor entires
 b = a * scale
 b = a * scale
 */
 */
 void _Scale(const XTensor * a, XTensor * b, int scale);
 void _Scale(const XTensor * a, XTensor * b, int scale);
+void _Scale(const XTensor * a, XTensor * b, float scale);
 
 
 /*
 /*
-scale tensor entires (on site)
+scale up tensor entires (on site)
 b = a * scale
 b = a * scale
 */
 */
 void Scale(XTensor & a, int scale);
 void Scale(XTensor & a, int scale);
+void Scale(XTensor & a, float scale);
     
     
 /*
 /*
-scale tensor entires
+scale up tensor entires
 b = a * scale
 b = a * scale
 */
 */
 void Scale(const XTensor & a, XTensor &b, int scale);
 void Scale(const XTensor & a, XTensor &b, int scale);
-
-//void Scale(const XTensor & a, XTensor & b, int num);
+void Scale(const XTensor & a, XTensor &b, float scale);
 
 
 /*
 /*
 descale tensor entires
 descale tensor entires