Changeset 21631e9

Timestamp:

Jan 28, 2019, 10:48:58 PM (5 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/crepe_org

Children:

1df9cd1

Parents:

75662eb

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);
   }