mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!4728 Fix tensor print format 

Merge pull request !4728 from hewei/fix_tensor_to_string
This commit is contained in:
mindspore-ci-bot 2020-08-20 08:52:07 +08:00 committed by Gitee
parent 8acb5bdf9b 3521f9e622
commit 213eacbd58
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/core/ir/tensor.cc
View File

@ -485,12 +485,12 @@ std::string Tensor::ToString() const {
auto dtype = Dtype();
MS_EXCEPTION_IF_NULL(dtype);
data_sync();
buf << "Tensor(shape=" << ShapeToString(shape_) << ", dtype=" << dtype->ToString() << ",\n";
buf << "Tensor(shape=" << ShapeToString(shape_) << ", dtype=" << dtype->ToString() << ',';
if (DataSize() < small_tensor_size) {
// Only print data for small tensor.
buf << data().ToString(data_type_, shape_) << ')';
buf << ((data().ndim() > 1) ? '\n' : ' ') << data().ToString(data_type_, shape_) << ')';
} else {
buf << "[...])";
buf << " [...])";
}
return buf.str();
}
@ -500,8 +500,8 @@ std::string Tensor::ToStringRepr() const {
auto dtype = Dtype();
MS_EXCEPTION_IF_NULL(dtype);
data_sync();
buf << "Tensor(shape=" << ShapeToString(shape_) << ", dtype=" << dtype->ToString() << ",\n"
<< data().ToString(data_type_, shape_) << ')';
buf << "Tensor(shape=" << ShapeToString(shape_) << ", dtype=" << dtype->ToString() << ','
<< ((data().ndim() > 1) ? '\n' : ' ') << data().ToString(data_type_, shape_) << ')';
return buf.str();
}