forked from huawei/mindspore2022
Add missing macro
This commit is contained in:
parent
a93476ba5d
commit
5a0a35ed47
|
|
@ -69,7 +69,7 @@ PYBIND_REGISTER(Tensor, 0, ([](const py::module *m) {
|
||||||
auto &tensor = py::cast<Tensor &>(t);
|
auto &tensor = py::cast<Tensor &>(t);
|
||||||
if (tensor.type() == DataType::DE_STRING) {
|
if (tensor.type() == DataType::DE_STRING) {
|
||||||
py::array res;
|
py::array res;
|
||||||
tensor.GetDataAsNumpyStrings(&res);
|
THROW_IF_ERROR(tensor.GetDataAsNumpyStrings(&res));
|
||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
py::buffer_info info;
|
py::buffer_info info;
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue