mat_mul.py (507B)
1 import random 2 import time 3 4 n = 400 5 6 A = [[random.random() for _ in range(n)] for _ in range(n)] 7 B = [[random.random() for _ in range(n)] for _ in range(n)] 8 9 C = [[0 for _ in range(n)] for _ in range(n)] 10 11 12 start = time.time() 13 14 for i in range(len(A)): 15 for k in range(len(B)): 16 for j in range(len(B)): 17 C[i][j] += A[i][k] * B[k][j] 18 19 20 end = time.time() 21 22 print(end - start) 23 24 # This runs in: 25 26 # 1: 13.906554698944092 27 # 2: 13.789746761322021 28 # 3: 14.16126012802124 29 30 # AVG: 13.952520529429117