python - 使用 Tensordot 复制矩阵-矩阵乘法

Question

在PyKeops package中，没有可用的 Matrix-Matrix 乘法公式。相反，他们实现了类似于numpy.tensordot. 我有两个A,B大小矩阵m x n和n x n. 有什么方法可以复制A @ B使用numpy.tensordot.

import numpy as np
m,n = 10, 20
A = np.random.random((m,n))
B = np.random.random((n,n))

result1 = A @ B
result2 = np.tensordot(A,B,(1,1))

print(result1 == result2)

# =======================
# array([[False, False, False, False, False, False, False, False, 
# ....

Answer 1

好的，我找到了答案，我们可以使用以下参数解决这个问题tensordot：

import numpy as np
m,n = 10, 20
A = np.random.random((m,n))
B = np.random.random((n,n))

result1 = A @ B
result2 = np.tensordot(A,B,[(1,),(0,)])

print(result1 == result2)

# =======================
# array([[True,  True,  True,  True,  True,  True,  True,  True,  True, 
# ....