wording

source/tensor/core/getandset/OnehotAndIndex.cpp

@@ -96,9 +96,12 @@ XTensor OnehotToIndex(const XTensor & onehot, int size)
 /* 
 convert index tensor to onehot tensor 
 
->> index - index tensor, which value is an integer num
->> onehot - onehot tensor, which value is 0 or 1
->> size - the last dimension size of the onehot tensor
+>> index - index of the output dimension (over the vocabulary)
+>> onehot - one-hot representation of the index
+>> size - vocabuary size (last dimension size of onehot)
+>> labelSmoothingP - the parameter that controls how smooth the output is.
+                     E.g., p = 0 means no smoothing
+                           p = 1 means a uniform distribution (almost)
 */
 void _IndexToOnehot(const XTensor * index, XTensor * onehot, 
                     int size, float labelSmoothingP)

source/tensor/core/movement/Gather.cpp

@@ -115,7 +115,7 @@ void _Gather(const XTensor * s, XTensor * t, XTensor * srcIndex)
 
         for (int i = 0; i < indexSize; i++) {
             int sIndex = sIndexData[i] * stride;
-            CheckNTErrors(sIndex < s->unitNum, "Wrong index!");
+            CheckNTErrors(sIndex < s->unitNum && sIndex >= 0, "Wrong index!");
             for (int j = 0; j < stride; j++)
                 tData[i * stride + j] = sData[sIndex + j];
         }

source/tensor/core/shape/Merge.cpp

@@ -185,13 +185,13 @@ bool CheckMergeSize(const XTensor * s, const XTensor * t, int whereToMerge, int
 transform a tensor by merging it along with a dimension (return an XTensor structure)
 make a new tensor to keep the result and  return it
 
-e.g., (N/3, M, 3) -> (N, M)
+e.g., (3, M, N/3) -> (M, N)
 
 >> s - the source tensor
 >> whereToMerge - the merging operation is along with which dimension
->> leadingDim - the leading dimension of merging, take (N/3, M, 3) -> (N, M) 
-   for example, whereToMerge = 0 (i.e., the dimension for "N/3")
-   leadingDim = 2 (i.e., the dimension for "3")
+>> leadingDim - the leading dimension of merging, take (3, M, N/3) -> (M, N) 
+   for example, whereToMerge = 2 (i.e., the dimension for "N/3")
+   leadingDim = 0 (i.e., the dimension for "3")
 << return - the transformed tensor by merging along with a dimension
 */
 XTensor Merge(const XTensor &s, int whereToMerge, int leadingDim)