Changeset f90051d

Timestamp:

Dec 29, 2021, 5:51:46 PM (2 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/cnn, feature/crepe

Children:

49e7171

Parents:

b496aa8

git-author:

Paul Brossier <piem@piem.org> (01/02/19 22:57:49)

git-committer:

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

Message:

[tensor] add have_same_size

File:

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

src/ai/tensor.c

@@ -111 +111 @@
 }
 
+uint_t aubio_tensor_have_same_size(aubio_tensor_t *t, aubio_tensor_t *s)
+{
+  uint_t n;
+  if (!t || !s) return 0;
+  if (t->ndim != s->ndim) return 0;
+  if (t->size != s->size) return 0;
+  n = t->ndim;
+  while (n--) {
+    if (t->shape[n] != s->shape[n]) {
+      return 0;
+    }
+  }
+  return 1;
+}
+
 smpl_t aubio_tensor_max(aubio_tensor_t *t)
 {