Changeset 50d7afe

Timestamp:

Jan 2, 2019, 10:47:51 PM (5 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/cnn_org, feature/crepe_org

Children:

40520ea

Parents:

124166d

Message:

[tensor] add get_shape_string

File:

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

src/ai/tensor.c

@@ -96 +96 @@
   return max;
 }
+
+const char_t *aubio_tensor_get_shape_string(aubio_tensor_t *t) {
+  uint_t i;
+  if (!t) return NULL;
+  size_t offset = 2;
+  static char_t shape_str[STRN_LENGTH];
+  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) ? "" : ", ");
+    strncpy(shape_str_previous, shape_str, len);
+  }
+  snprintf(shape_str, strnlen(shape_str, STRN_LENGTH - offset - 1) + offset,
+      "%s)", shape_str_previous);
+}