在非常大的矩阵上计算点积

Question

如何计算M1形状5e7x50和M2形状50x5e7的点积？还可以将一个矩阵5e7x5e7除以另一个大小相同的矩阵吗？

是要走的路吗？

Answer 1

您可以在GPU中使用tensorflow：

import numpy as np
import tensorflow as tf

tf.matmul(np.array(....), np.array(....))

Answer 2

尝试使用numpy：

np.dot(mat_a, mat_b)

如果按除法表示的是按元素划分，则还可以使用：

np.divide(x1, x2)

Answer 3

您可以在numpy中使用@运算符。

import numpy as np

np.array(...) @ np.array(...)