Reshape autograd

2024-05-01 12:11:50 -03:00 · 2024-05-01 12:11:50 -03:00 · 769666cfa8
commit 769666cfa8
parent 9981bb8b6f
15 changed files with 83 additions and 35 deletions
--- a/norch/csrc/cpu.cpp
+++ b/norch/csrc/cpu.cpp
@ -86,3 +86,10 @@ void transpose_tensor_cpu(Tensor* tensor, float* result_data) {
        }
    }
 }
+
+void assign_tensor_cpu(Tensor* tensor, float* result_data) {
+
+    for (int i = 0; i < tensor->size; i++) {
+        result_data[i] = tensor->data[i];
+    }
+}