计算超大矩阵上的点积

Question

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

细分是一条路吗

Answer 1

尝试使用numpy：

np.dot(mat_a, mat_b)

如果“除法”是指元素除法，则也可以使用：

np.divide(x1, x2)

Answer 2

可以在numpy中使用@运算符

import numpy as np

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

Answer 3

您可以将tensorflow与GPU一起使用：

import numpy as np
import tensorflow as tf

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