Changeset e7779f0 for src/ai/tensor.c

Timestamp:

Dec 29, 2021, 5:52:00 PM (3 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/crepe

Children:

461a89f

Parents:

7f270f8

git-author:

Paul Brossier <piem@piem.org> (01/28/19 22:48:58)

git-committer:

Paul Brossier <piem@piem.org> (12/29/21 17:52:00)

Message:

[tensor] improve shape string for 1d

File:

• src/ai/tensor.c (modified) (1 diff)

src/ai/tensor.c

@@ -170 +170 @@
   char_t shape_str_previous[STRN_LENGTH] = "(";
   for (i = 0; i < t->ndim; i++) {
-    int len = snprintf(shape_str, STRN_LENGTH, "%s%d%s",
-        shape_str_previous, t->shape[i], (i == t->ndim - 1) ? "" : ", ");
+    // and space last if not the last one
+    int add_space = (i < t->ndim - 1);
+    // add coma first if this not last, or always if 1d
+    int add_coma = add_space || (t->ndim == 1);
+    int len = snprintf(shape_str, STRN_LENGTH, "%s%d%s%s",
+        shape_str_previous, t->shape[i],
+        add_coma ? "," : "", add_space ? " " : "");
     strncpy(shape_str_previous, shape_str, len);
   }