fix a ndarray dim vector condition

This commit is contained in:
Qianqian Fang 2022-02-28 12:49:58 -05:00
parent 5786991973
commit b54f058dcb
2 changed files with 2 additions and 2 deletions

View File

@ -2112,7 +2112,7 @@ class binary_reader
{
return false;
}
if (dim.size() == 1 || (dim.size() > 1 && dim.at(0) == 1)) // return normal array size if 1D row vector
if (dim.size() == 1 || (dim.size() == 2 && dim.at(0) == 1)) // return normal array size if 1D row vector
{
result = dim.at(dim.size() - 1);
return true;

View File

@ -10488,7 +10488,7 @@ class binary_reader
{
return false;
}
if (dim.size() == 1 || (dim.size() > 1 && dim.at(0) == 1)) // return normal array size if 1D row vector
if (dim.size() == 1 || (dim.size() == 2 && dim.at(0) == 1)) // return normal array size if 1D row vector
{
result = dim.at(dim.size() - 1);
return true;