|
|
@@ -553,7 +553,7 @@ std::string Tensor::ToStringInternal(int limit_size) const { |
|
|
std::ostringstream buf; |
|
|
std::ostringstream buf; |
|
|
auto dtype = Dtype(); |
|
|
auto dtype = Dtype(); |
|
|
MS_EXCEPTION_IF_NULL(dtype); |
|
|
MS_EXCEPTION_IF_NULL(dtype); |
|
|
buf << "Tensor(id=" << id_ << ", shape=" << ShapeToString(shape_) << ", dtype=" << dtype->ToString() << ", value="; |
|
|
|
|
|
|
|
|
buf << "Tensor(shape=" << ShapeToString(shape_) << ", dtype=" << dtype->ToString() << ", value="; |
|
|
if (limit_size <= 0 || DataSize() < limit_size) { |
|
|
if (limit_size <= 0 || DataSize() < limit_size) { |
|
|
// Only print data for small tensor. |
|
|
// Only print data for small tensor. |
|
|
buf << ((data().ndim() > 1) ? '\n' : ' ') << data().ToString(data_type_, shape_, false); |
|
|
buf << ((data().ndim() > 1) ? '\n' : ' ') << data().ToString(data_type_, shape_, false); |
|
|
|