p85971406
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!48186 Enable dynamic length sequence to be convert to python object. 

Merge pull request !48186 from LiangZhibo/dynamic
This commit is contained in:
i-robot 2023-02-02 07:32:26 +00:00 committed by Gitee
parent 6500d2afd8 a359933a4d
commit 368425fce4
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mindspore/ccsrc/utils/convert_utils_py.cc
View File

@ -472,6 +472,22 @@ py::object VectorRefToPyData(const VectorRef &value_list, const AbstractBasePtr
} }
auto seq_abs = abs->cast<abstract::AbstractSequencePtr>(); auto seq_abs = abs->cast<abstract::AbstractSequencePtr>();
MS_EXCEPTION_IF_NULL(seq_abs); MS_EXCEPTION_IF_NULL(seq_abs);
bool dynamic_len = seq_abs->dynamic_len();
auto dynamic_len_element_abs = seq_abs->dynamic_len_element_abs();
if (dynamic_len || dynamic_len_element_abs != nullptr) {
if (dynamic_len_element_abs == nullptr) {
MS_LOG(INFO) << "Dynamic length sequence with no specified element abstract convert to empty tuple.";
return ref_tuple;
}
if (dynamic_len_element_abs->isa<abstract::AbstractNone>()) {
MS_LOG(INFO) << "Dynamic length sequence with element None convert to empty tuple.";
return ref_tuple;
}
for (size_t i = 0; i < value_size; ++i) {
ref_tuple[i] = BaseRefToPyData(value_list[i], dynamic_len_element_abs);
}
return ref_tuple;
}
// The size of seq_abs may be larger than the size of value_list, because the backend will eliminate None. // The size of seq_abs may be larger than the size of value_list, because the backend will eliminate None.
size_t ref_idx = 0; size_t ref_idx = 0;
for (size_t i = 0; i < seq_abs->size(); i++) { for (size_t i = 0; i < seq_abs->size(); i++) {