From ad22eda87c8c519ca2d9809533e63a3d7b34b71c Mon Sep 17 00:00:00 2001
From: gingerBill <bill@gingerbill.org>
Date: Mon, 28 Jun 2021 13:23:28 +0100
Subject: Improve `linalg.transpose` type determination for square matrices

---
 core/math/linalg/general.odin | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/core/math/linalg/general.odin b/core/math/linalg/general.odin
index d0c558758..318605ccf 100644
--- a/core/math/linalg/general.odin
+++ b/core/math/linalg/general.odin
@@ -170,8 +170,7 @@ trace :: proc(m: $T/[$N][N]$E) -> (tr: E) {
 	return;
 }
 
-
-transpose :: proc(a: $T/[$N][$M]$E) -> (m: [M][N]E) {
+transpose :: proc(a: $T/[$N][$M]$E) -> (m: (T when N == M else [M][N]E)) {
 	for j in 0..<M {
 		for i in 0..<N {
 			m[j][i] = a[i][j];
-- 
cgit v1.2.3