python各框架和模块对矩阵的运算比较

最近在学习numpy，pandas，tensorflow和pytorch，突然想试试各种不同的方法对矩阵的运算效率如何，以下是代码：

import tensorflow as tf
import pandas as pd
import torch
import time
import numpy as np
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'#mac Os 下有一个警告可以通过这个语句使之不显

print('==============TensorFlow==============')

mat_1 = tf.constant(np.arange(12).reshape(3,4).astype(np.float32))
mat_2 = tf.constant(np.arange(12).reshape(4,3).astype(np.float32))

start = time.time()
product = tf.matmul(mat_1,mat_2)
sess = tf.Session()
res = sess.run(product)
print(res)
end = time.time()
print('run rime = ',end-start)
print()


print('==============PyTorch==============')

x = torch.from_numpy(np.arange(12).reshape(3,4).astype(np.float32))
y = torch.from_numpy(np.arange(12).reshape(4,3).astype(np.float32))

start = time.time()
z = torch.mm(x,y)
print(z)
end = time.time()
print('run rime = ',end-start)
print()


print('==============Pandas==============')

x = pd.DataFrame(np.arange(12).reshape(3,4).astype(np.float32))
y = pd.DataFrame(np.arange(12).reshape(4,3).astype(np.float32))

start = time.time()
z = x.dot(y)
print(z)
end = time.time()
print('run rime = ',end-start)
print()



print('==============NumPy==============')

x = np.arange(12).reshape(3,4).astype(np.float32)
y = np.arange(12).reshape(4,3).astype(np.float32)

start = time.time()
z = x.dot(y)
print(z)
end = time.time()
print('run rime = ',end-start)
print()

以下是运行结果：

可以看到pytorch确实比tensorflow对矩阵的运算效率要高不少。