Commit b4355def by liyinqiao

Merge with ZHANG Yuhao branch (Don't use this! It's an incomplete version)

Fix the dimension bugs in Stack functions.
parent e84c0f66
...@@ -95,7 +95,7 @@ XTensor Stack(const TensorList &smalls, int dim) ...@@ -95,7 +95,7 @@ XTensor Stack(const TensorList &smalls, int dim)
if (i < dim) if (i < dim)
dimSize[i] = tensor->GetDim(i); dimSize[i] = tensor->GetDim(i);
else if (i > dim) else if (i > dim)
dimSize[i] = tensor->GetDim(i); dimSize[i] = tensor->GetDim(i - 1);
else if (i == dim) else if (i == dim)
dimSize[i] = count; dimSize[i] = count;
} }
...@@ -160,7 +160,7 @@ void Stack(const TensorList &smalls, XTensor &t, int dim) ...@@ -160,7 +160,7 @@ void Stack(const TensorList &smalls, XTensor &t, int dim)
if (i < dim) if (i < dim)
dimSize[i] = tensor->GetDim(i); dimSize[i] = tensor->GetDim(i);
else if (i > dim) else if (i > dim)
dimSize[i] = tensor->GetDim(i-1); dimSize[i] = tensor->GetDim(i - 1);
else if (i == dim) else if (i == dim)
dimSize[i] = count; dimSize[i] = count;
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论