Commit 287e226c by xiaotong

make XTensor::GetDim support negtive index

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