Improve linalg.transpose type determination for square matrices

This commit is contained in:
gingerBill
2021-06-28 13:23:28 +01:00
parent 8d31ba492d
commit ad22eda87c

View File

@@ -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];