Changeset 6006760 for src

Timestamp:

Jan 7, 2019, 11:26:09 PM (6 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/cnn_org, feature/crepe_org

Children:

791b436

Parents:

147afba

Message:

[tensor] add matmul

Location:

src/ai

Files:

• tensor.c (modified) (1 diff)
• tensor.h (modified) (1 diff)

src/ai/tensor.c

@@ -204 +204 @@
   AUBIO_MSG("\n");
 }
+
+void aubio_tensor_matmul(aubio_tensor_t *a, aubio_tensor_t *b,
+    aubio_tensor_t *c)
+{
+  AUBIO_ASSERT (a->shape[0] == c->shape[0]);
+  AUBIO_ASSERT (a->shape[1] == b->shape[0]);
+  AUBIO_ASSERT (b->shape[1] == c->shape[1]);
+#if !defined(HAVE_BLAS)
+  uint_t i, j, k;
+  for (i = 0; i < c->shape[0]; i++) {
+    for (j = 0; j < c->shape[1]; j++) {
+      smpl_t sum = 0.;
+      for (k = 0; k < a->shape[1]; k++) {
+          sum += a->buffer[i * a->shape[1] + k]
+            * b->buffer[k * b->shape[1] + j];
+      }
+      c->buffer[i * c->shape[1] + j] = sum;
+    }
+  }
+#else
+  cblas_sgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, a->shape[0],
+      b->shape[1], b->shape[0], 1.F, a->buffer, a->shape[1], b->buffer,
+      b->shape[1], 0.F, c->buffer, b->shape[1]);
+#endif
+}

src/ai/tensor.h

@@ -149 +149 @@
 const char_t *aubio_tensor_get_shape_string(aubio_tensor_t *t);
 
+void aubio_tensor_matmul(aubio_tensor_t *a, aubio_tensor_t *b,
+    aubio_tensor_t *c);
+
 #ifdef __cplusplus
 }