diff --git a/mindspore/ccsrc/utils/tensorprint_utils.cc b/mindspore/ccsrc/utils/tensorprint_utils.cc index b34944cf215442f8a01e52c5af1c1676cacdd4a7..56f0f13f4b9a354e56c42c605189c5fb626e84d5 100644 --- a/mindspore/ccsrc/utils/tensorprint_utils.cc +++ b/mindspore/ccsrc/utils/tensorprint_utils.cc @@ -32,6 +32,7 @@ namespace mindspore { const char kShapeSeperator[] = ","; const char kShapeScalar[] = "[0]"; +const char kShapeNone[] = "[]"; static std::map print_type_map = { {"int8_t", TypeId::kNumberTypeInt8}, {"uint8_t", TypeId::kNumberTypeUInt8}, {"int16_t", TypeId::kNumberTypeInt16}, {"uint16_t", TypeId::kNumberTypeUInt16}, @@ -163,9 +164,9 @@ bool ConvertDataItem2Tensor(const std::vector &items) { } std::shared_ptr str_data_ptr = std::static_pointer_cast(item.dataPtr_); MS_EXCEPTION_IF_NULL(str_data_ptr); - if (item.tensorShape_ == kShapeScalar) { + if (item.tensorShape_ == kShapeScalar || item.tensorShape_ == kShapeNone) { if (!judgeLengthValid(str_data_ptr->size(), item.tensorType_)) { - MS_LOG(EXCEPTION) << "Print op receive data length is invalid."; + MS_LOG(EXCEPTION) << "Print op receive data length is invalid."; } convertDataItem2Scalar(str_data_ptr->data(), item.tensorType_, &buf); continue;