From 3a4ecebcfa3a3650da61c1fc28476a5309c36bf3 Mon Sep 17 00:00:00 2001
From: Elias Kohout <elias@kohout.de>
Date: Thu, 7 Sep 2023 13:48:56 +0200
Subject: [PATCH] Fixed memory leak.

---
 tensor.c | 30 ++++++++++++++++++++++++------
 1 file changed, 24 insertions(+), 6 deletions(-)

diff --git a/tensor.c b/tensor.c
index e729ec2..8888dc7 100644
--- a/tensor.c
+++ b/tensor.c
@@ -361,8 +361,14 @@ tensor tensor_add(const tensor t1, const tensor t2)
 
 	tensor t3 = tensor_new();
 	if(t3 == NULL) return NULL;
-	if (!tensor_cpy(t3, t1)) return NULL;
-	if (!tensor_add_inplace(t3, t2)) return NULL;
+	if (!tensor_cpy(t3, t1)) {
+        tensor_destroy(t3);
+        return NULL;
+    }
+	if (!tensor_add_inplace(t3, t2)) {
+        tensor_destroy(t3);
+        return NULL;
+    }
 	return t3;
 }
 
@@ -381,8 +387,14 @@ tensor tensor_sub(const tensor t1, const tensor t2)
 
 	tensor t3 = tensor_new();
 	if(t3 == NULL) return NULL;
-	if (!tensor_cpy(t3, t1)) return NULL;
-	if (!tensor_sub_inplace(t3, t2)) return NULL;
+	if (!tensor_cpy(t3, t1)) {
+        tensor_destroy(t3);
+        return NULL;
+    }
+	if (!tensor_sub_inplace(t3, t2)) {
+        tensor_destroy(t3);
+        return NULL;
+    }
 	return t3;
 }
 
@@ -402,8 +414,14 @@ tensor tensor_mul(const tensor t1, const tensor t2)
 
 	tensor t3 = tensor_new();
 	if(t3 == NULL) return NULL;
-	if (!tensor_cpy(t3, t1)) return NULL;
-	if (!tensor_mul_inplace(t3, t2)) return NULL;
+	if (!tensor_cpy(t3, t1)) {
+        tensor_destroy(t3);
+        return NULL;
+    }
+	if (!tensor_mul_inplace(t3, t2)) {
+        tensor_destroy(t3);
+        return NULL;
+    }
 	return t3;
 }