From f90051d8f440b3aa8c0bf8e90fba9968764b9442 Mon Sep 17 00:00:00 2001
From: Paul Brossier <piem@piem.org>
Date: Wed, 2 Jan 2019 22:57:49 +0100
Subject: [PATCH] [tensor] add have_same_size

---
 src/ai/tensor.c | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/ai/tensor.c b/src/ai/tensor.c
index efa4210e..b52418f0 100644
--- a/src/ai/tensor.c
+++ b/src/ai/tensor.c
@@ -110,6 +110,21 @@ uint_t aubio_tensor_get_subtensor(aubio_tensor_t *t, uint_t i,
   return AUBIO_OK;
 }
 
+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)
 {
   uint_t i;
-- 
2.11.0