Changeset a33c395 for src/ai/tensor.c

Timestamp:

Dec 29, 2021, 5:51:48 PM (3 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/cnn, feature/crepe

Children:

6d1566e

Parents:

1f13e29

git-author:

Paul Brossier <piem@piem.org> (01/07/19 23:26:09)

git-committer:

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

Message:

[tensor] add matmul

File:

• src/ai/tensor.c (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
+}