Changeset cc74a29

Timestamp:

Jan 29, 2019, 1:14:21 AM (5 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/crepe_org

Children:

3ff9654

Parents:

3ac1426

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