fix a ndarray dim vector condition
This commit is contained in:
parent
5786991973
commit
b54f058dcb
@ -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;
|
||||
|
||||
@ -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;
|
||||
|
||||
Loading…
Reference in New Issue
Block a user