-
Notifications
You must be signed in to change notification settings - Fork 22
/
Copy pathmatmul.k
110 lines (98 loc) · 1.12 KB
/
matmul.k
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
matmul: {
A::x
B::y
n::#A // no. of rows of A
m::#*A // no. of col of A
p::#*B // no. of cols of B
C::(n;p)#0
i::0
j::0
k::0
sum::0
{
i::x
{
j::x
sum::0
{
k::x
sum::sum+A[i;k]*B[k;j]
}'!m
C[i;j]::sum
}'!p
}'!n
Ci}
/ Removing sum
matmul: {
A::x
B::y
n::#A
m::#*A
p::#*B
C::(n;p)#0
i::0
j::0
k::0
{
i::x
{
j::x
C[i;j]::+/{
k::x
A[i;k]*B[k;j]
}'!m
}'!p
}'!n
C}
/ Removing C
matmul: {
A::x
B::y
n::#A
m::#*A
p::#*B
i::0
j::0
k::0
{
i::x
{
j::x
+/{
k::x
A[i;k]*B[k;j] }'!m }'!p }'!n}
/ removing k and m
matmul: {
A::x
B::y
n::#A
p::#*B
i::0
j::0
{
i::x
{
j::x
+/A[i]*B[;j] }'!p }'!n}
/ removing j and p
matmul: {
A::x
B::y
n::#A
i::0
{
i::x
A[i]{+/x*y}/:+B}'!n }
/ removing i and n
matmul: {
A::x
B::y
A{+/x*y}/:\:+B }
/ removing A and B
matmul: {x{+/x*y}/:\:+y}
/ remove the transpose
matmul: {x{+/x*y}\:y}
/ tacit
matmul: {x(+/*)\:y}
matmul: (+/*)\:
matmul[3 3#!9;=3]