Browse Source

!15484 fix tensor print

From: @huangbingjian
Reviewed-by: @zh_qh,@ginfung
Signed-off-by: @zh_qh
pull/15484/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
0b8d4348f6
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/core/ir/tensor.cc

+ 1
- 1
mindspore/core/ir/tensor.cc View File

@@ -367,7 +367,7 @@ class TensorDataImpl : public TensorData {
if (num > kThreshold / 2) {
ssize_t iter_times = std::min(static_cast<ssize_t>(num - kThreshold / 2), static_cast<ssize_t>(kThreshold / 2));
for (ssize_t i = 0; i < iter_times; i++) {
if (use_comma && i != 0) {
if (use_comma && (i != 0 || num <= kThreshold)) { // Not just after ignored part || Not handle ignored part
ss << ',';
}
ss << '\n';


Loading…
Cancel
Save