利用Tensordot复制矩阵-矩阵乘法

Question

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

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, 
# ....