Browse Source

Sync data while Tensor::ToString()

tags/v0.7.0-beta
He Wei 5 years ago
parent
commit
f9ab2781ae
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/core/ir/tensor.cc

+ 2
- 0
mindspore/core/ir/tensor.cc View File

@@ -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