Changeset c4b6b59

Timestamp:

Jan 2, 2019, 10:58:45 PM (6 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/cnn_org, feature/crepe_org

Children:

1343aca

Parents:

dc257cc

Message:

[tensor] add print helpers

File:

• src/ai/tensor.c (modified) (2 diffs)

src/ai/tensor.c

@@ -2 +2 @@
 #include "fmat.h"
 #include "tensor.h"
+
+#define STRN_LENGTH 40
+#if !HAVE_AUBIO_DOUBLE
+#define AUBIO_SMPL_TFMT "% 9.4f"
+#else
+#define AUBIO_SMPL_TFMT "% 9.4lf"
+#endif /* HAVE_AUBIO_DOUBLE */
 
 aubio_tensor_t *new_aubio_tensor(uint_t ndim, uint_t *shape)
@@ -151 +158 @@
   return shape_str;
 }
+
+static void aubio_tensor_print_subtensor(aubio_tensor_t *t, uint_t depth)
+{
+  uint_t i;
+  AUBIO_MSG("[");
+  for (i = 0; i < t->shape[0]; i ++) {
+    AUBIO_MSG("%*s", i == 0 ? 0 : depth + 1, i == 0 ? "" : " ");
+    if (t->ndim == 1) {
+      AUBIO_MSG(AUBIO_SMPL_TFMT, t->buffer[i]);
+    } else {
+      aubio_tensor_t st;
+      aubio_tensor_get_subtensor(t, i, &st);
+      aubio_tensor_print_subtensor(&st, depth + 1); // recursive call
+    }
+    AUBIO_MSG("%s%s", (i < t->shape[0] - 1) ? "," : "",
+        t->ndim == 1 ? " " : ((i < t->shape[0] - 1) ? "\n" : ""));
+  }
+  AUBIO_MSG("]");
+}
+
+void aubio_tensor_print(aubio_tensor_t *t)
+{
+  AUBIO_MSG("tensor of shape %s\n", aubio_tensor_get_shape_string(t));
+  aubio_tensor_print_subtensor(t, 0);
+  AUBIO_MSG("\n");
+}