Commit 287e226c by xiaotong

make XTensor::GetDim support negtive index

parent c726938f
...@@ -427,7 +427,11 @@ int XTensor::GetDim(const int dim) ...@@ -427,7 +427,11 @@ int XTensor::GetDim(const int dim)
{ {
CheckNTErrors(dim < order, "dimenision is out of range!"); CheckNTErrors(dim < order, "dimenision is out of range!");
return dimSize[dim]; int d = dim;
if(dim < 0)
d = order - 1;
return dimSize[d];
} }
/* /*
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论