matmul.py

import tensorflow as tf
import time
m1 = []
m2 = []
result = []
i = 10
while i <= 10000:
    m1.append(tf.random_uniform(shape = [i, i]))
    m2.append(tf.random_uniform(shape = [i, i]))
    i *= 10
for t1, t2 in zip(m1, m2):
    result.append(tf.matmul(t1, t2))
sess = tf.Session()
for tensor in result:
    start = time.time()
    sess.run(tensor)
    print(tensor.shape, time.time() - start)