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