Add SetDataFixedInt

source/tensor/core/getandset/SetData.cpp

@@ -168,6 +168,17 @@ void SetDataFixed(XTensor &tensor, DTYPE p)
 {
 {
     _SetDataFixed(&tensor, &p);
     _SetDataFixed(&tensor, &p);
 }
 }
+    
+/*
+generate data items with a fixed value p (in integer)
+>> tensor - the tensor whose data array would be initialized
+>> p - an integer
+*/
+void SetDataFixedInt(XTensor &tensor, int p)
+{
+    CheckNTErrors(tensor.dataType == X_INT, "An integer tensor is required!");
+    _SetDataFixed(&tensor, &p);
+}
 
 
 /* 
 /* 
 generate data items with a fixed value p (in integer) 
 generate data items with a fixed value p (in integer) 

source/tensor/core/getandset/SetData.h

@@ -35,6 +35,9 @@ void _SetDataFixed(XTensor * tensor, void * valuePointer);
 
 
 /* generate data items with a fixed value p (in default type) */
 /* generate data items with a fixed value p (in default type) */
 void SetDataFixed(XTensor &tensor, DTYPE p);
 void SetDataFixed(XTensor &tensor, DTYPE p);
+    
+/* generate data items with a fixed value p (in integer) */
+void SetDataFixedInt(XTensor &tensor, int p);
 
 
 /* generate data items with a fixed value p (in int) */
 /* generate data items with a fixed value p (in int) */
 void _SetDataFixedInt(XTensor * tensor, int p);
 void _SetDataFixedInt(XTensor * tensor, int p);
@@ -72,4 +75,4 @@ void _SetDataWithOffsetAndValue(XTensor * tensor, MTYPE * offsets, void * values
 
 
 } // namespace nts(NiuTrans.Tensor)
 } // namespace nts(NiuTrans.Tensor)
 
 
-#endif // __SETDATA_H__
+#endif // __SETDATA_H__
\ No newline at end of file