Changeset 0dad503 for src/ai

Timestamp:

Dec 29, 2021, 5:52:01 PM (3 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/crepe

Children:

1b9fe24

Parents:

7897b51

git-author:

Paul Brossier <piem@piem.org> (01/29/19 01:14:21)

git-committer:

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

Message:

[tensor] add copy

Location:

src/ai

Files:

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

src/ai/tensor.c

@@ -235 +235 @@
 #endif
 }
+
+void aubio_tensor_copy(aubio_tensor_t *s, aubio_tensor_t *t)
+{
+  if (!aubio_tensor_have_same_shape(s, t)) {
+    AUBIO_ERR("tensor: not copying source tensor %s",
+        aubio_tensor_get_shape_string(s));
+    AUBIO_ERR(" to destination tensor %s",
+        aubio_tensor_get_shape_string(t));
+    return;
+  }
+  AUBIO_MEMCPY(t->buffer, s->buffer, s->size);
+}

src/ai/tensor.h

@@ -160 +160 @@
 const char_t *aubio_tensor_get_shape_string(aubio_tensor_t *t);
 
+/** compute matrix multiplication A x B = C
+
+  \param a  left tensor
+  \param b  right tensor
+  \param c  output tensor
+
+*/
 void aubio_tensor_matmul(aubio_tensor_t *a, aubio_tensor_t *b,
     aubio_tensor_t *c);
+
+/** copy tensor
+
+  \param s  source tensor
+  \param d  destination tensor
+
+*/
+void aubio_tensor_copy(aubio_tensor_t *s, aubio_tensor_t *d);
 
 #ifdef __cplusplus