!4206 Fix Tensor::ToString() data sync problem

Merge pull request !4206 from hewei/fix_tensor_to_string
pull/4206/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit a0531b7b41

@ -465,6 +465,7 @@ std::string Tensor::GetShapeAndDataTypeInfo() const {
std::string Tensor::ToString() const {
const int small_tensor_size = 30;
std::ostringstream buf;
data_sync();
buf << "Tensor shape:[" << shape() << "]" << this->Dtype()->ToString();
// only print small tensor
if (DataSize() < small_tensor_size) {
@ -477,6 +478,7 @@ std::string Tensor::ToStringRepr() const {
std::ostringstream buf;
auto type_ptr = this->Dtype();
MS_EXCEPTION_IF_NULL(type_ptr);
data_sync();
buf << "Tensor shape:[" << shape() << "]" << type_ptr->ToString();
buf << "\nvalue:" << data().ToString(data_type_, shape());
return buf.str();

Loading…
Cancel
Save