Changeset 17189b2 for src/ai/tensor.c

Timestamp:

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

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/crepe

Children:

3c754d5

Parents:

461a89f

git-author:

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

git-committer:

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

Message:

[tensor] rewrite and rename have_same_shape

File:

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

src/ai/tensor.c

@@ -138 +138 @@
 }
 
-uint_t aubio_tensor_have_same_size(aubio_tensor_t *t, aubio_tensor_t *s)
+uint_t aubio_tensor_have_same_shape(aubio_tensor_t *a, aubio_tensor_t *b)
 {
   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]) {
+  AUBIO_ASSERT(a && b);
+  if (a->ndim != b->ndim) {
+    return 0;
+  }
+  for (n = 0; n < a->ndim; n++) {
+    if (a->shape[n] != b->shape[n]) {
       return 0;
     }